linux配置cuda环境
时间: 2025-07-19 17:01:54 浏览: 3
### 如何在Linux系统上配置CUDA环境
#### 安装前准备
为了确保顺利安装CUDA工具包,建议先确认系统的GCC版本。虽然运行CUDA应用不依赖于特定的GCC版本,但在开发过程中需要合适的GCC版本[^3]。
#### 添加CUDA仓库并设置优先级
对于基于Debian/Ubuntu的发行版,可以通过创建pin文件来提高CUDA官方源的优先级:
```bash
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
```
这一步骤有助于防止软件包管理器自动更新覆盖掉已安装好的CUDA版本[^2]。
#### 配置APT源列表
接着需向`/etc/apt/sources.list.d/`目录下添加相应的CUDA APT源地址。具体命令如下所示(假设目标操作系统为Ubuntu 18.04 LTS):
```bash
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
echo 'deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /' | sudo tee /etc/apt/sources.list.d/cuda.list
```
完成上述操作之后即可执行标准的APT刷新流程以获取最新的可用包信息:
```bash
sudo apt-get update
```
#### 安装CUDA Toolkit及相关驱动程序
现在可以正式开始安装CUDA Toolkit以及必要的NVIDIA显卡驱动了:
```bash
sudo apt-get install cuda-drivers cuda-toolkit-10-2
```
这里选择了具体的CUDA版本号作为例子;实际部署时可根据需求调整至其他稳定发布的版本。
#### 设置环境变量
为了让当前shell会话能够识别新安装的CUDA路径,推荐编辑用户的`.bashrc`或全局的`/etc/profile`文件加入以下两行内容:
```bash
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
使更改生效的方法是在终端里输入source ~/.bashrc 或者重新启动计算机。
#### 卸载CUDA Toolkit (如果需要重装或其他原因)
当遇到问题想要彻底清除现有安装以便重新开始时,可利用yum命令移除所有与CUDA有关联的数据包:
```bash
sudo yum remove "*cublas*" "cuda*"
```
此过程将会删除所有的CUBLAS库和CUDA组件,从而达到完全卸载的效果[^1]。
阅读全文
相关推荐


















