如何在conda虚拟环境中设置CUDA_HOME变量

原创已于 2025-10-16 16:24:35 修改 · 278 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#conda #python #深度学习

于 2025-10-16 16:23:42 首次发布

如果是使用系统的cuda，那么方法比较简单，直接：

export CUDA_HOME=/usr/local/cuda

如果是使用conda虚拟环境中下载的cuda，首先确保cuda和cudnn确实安装了：

print(torch.version.cuda)
print(torch.backends.cudnn.version())

然后安装

conda install -c conda-forge cudatoolkit-dev -y

查看是否成功安装：

which nvcc
nvcc --version

最后设置路径：

export CUDA_HOME=$CONDA_PREFIX

如果希望每次激活conda环境都自动设置该环境变量，可以参考以下步骤：

# 在虚拟环境中创建激活脚本文件夹（如果还没有的话）：
mkdir -p $CONDA_PREFIX/etc/conda/activate.d

# 在该目录下新建一个脚本文件，比如env_vars.sh，内容如下：
#!/bin/bash
export CUDA_HOME=$CONDA_PREFIX

# 给脚本文件授予可执行权限：
chmod +x $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh

参考：
https://vinesmsuic.github.io/linux-conda-cudahome/index.html

https://blog.youkuaiyun.com/qq_43705697/article/details/121618276

https://blog.youkuaiyun.com/qq_45445740/article/details/145810428