CUDA环境变量如何设置
时间: 2024-12-02 18:21:57 浏览: 64
CUDA环境变量的设置取决于你的操作系统(Ubuntu、CentOS、Windows/Linux Subsystem for Windows等)。以下是几个常见的操作系统设置步骤:
**对于Linux (Ubuntu/Debian)**:
1. **添加CUDA repositories**: 在终端中输入:
```
sudo apt-get install nvidia-cuda-toolkit
```
2. **设置环境变量**: 打开`.bashrc`或`.bash_profile`文件,然后添加类似这样的行:
```
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
保存并关闭文件,再运行 `source ~/.bashrc` 或 `source ~/.bash_profile` 使更改生效。
**对于CentOS/RHEL:**
1. 安装CUDA:
```
sudo yum install epel-release -y
sudo yum install cuda
```
2. 设置环境变量,类似于Linux Ubuntu部分。
**对于Windows/Linux Subsystem for Windows (WSL):**
1. 双击打开PowerShell作为管理员。
2. 添加路径到环境变量:
```powershell
$env:Path += ";/mnt/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.0/bin"
$env:Path += ";/mnt/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.0/extras/CUPTI/lib/x64"
```
3. 将`CUDA_HOME`指向CUDA安装路径:
```powershell
$env:CUDA_HOME = "/mnt/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.0"
```
确保上述操作完成后,你就可以在命令行中通过`nvcc`等命令来验证CUDA是否已成功启用。
阅读全文
相关推荐


















