服务器设置CUDA HOME环境变量
时间: 2025-01-19 07:41:37 浏览: 137
### 设置 CUDA_HOME 环境变量
为了确保 GPU 加速的应用程序能够正常工作,在 Linux 服务器环境中正确设置 `CUDA_HOME` 和其他相关环境变量至关重要。当遇到错误提示 `AssertionError: CUDA_HOME does not exist, unable to compile CUDA op(s)`[^1],这通常意味着系统无法找到 CUDA 的安装路径。
#### 修改 `.bashrc` 文件添加环境变量
对于大多数基于 Debian/Ubuntu 的发行版来说,推荐通过编辑用户的 shell 配置文件(通常是 `~/.bashrc` 或者 `/etc/profile.d/cuda.sh`)来永久设定这些变量:
```bash
export CUDA_HOME=/usr/local/cuda-11.4 # 根据实际安装位置调整此路径
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_HOME/lib64
```
上述命令假设已经完成了 NVIDIA 驱动以及特定版本的 CUDA 工具包 (如 CUDA 11.4)[^2] 安装过程,并且希望将其作为默认使用的 CUDA 版本。
完成编辑之后,可以通过执行以下命令使更改立即生效:
```bash
source ~/.bashrc
```
或者重启终端会话也可以达到相同的效果。
验证设置是否成功的方法之一就是尝试调用 nvcc 编译器并查看其版本号:
```bash
nvcc --version
```
如果一切顺利,则应该能看到所期望的 CUDA 版本信息被打印出来。
另外需要注意的是,如果有多个不同版本的 PyTorch 或者其他的依赖于特定 CUDA 版本的库存在时,可能还需要进一步确认各个软件包之间兼容性的关系[^4]。
阅读全文
相关推荐


















