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

如果是使用系统的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.csdn.net/qq_43705697/article/details/121618276

https://blog.csdn.net/qq_45445740/article/details/145810428