Ubuntu 给虚拟环境指定CUDA版本

具体参考： https://zhuanlan.zhihu.com/p/137874101 （如果侵权，速删）
Ubuntu安装了多个版本的CUDA，不同的虚拟环境需要切换不同的CUDA
本文以CUDA10.2和CUDA11.2为例：

1.首先激活虚拟环境 pytorch，输出当前环境的路径

conda activate pytorch
echo ${CONDA_PREFIX}

得到路径，例如：/home/username/anaconda3/envs/pytorch

2.activate文件夹

#建立激活环境下的文件夹
mkdir -p /home/username/anaconda3/envs/pytorch/etc/conda/activate.d
#写入脚本
vi /home/username/anaconda3/envs/pytorch/etc/conda/activate.d/activate.sh

添加如下内容：

ORIGINAL_CUDA_HOME=$CUDA_HOME
ORIGINAL_LD_LIBRARY_PATH=$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-11.2
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

3.deactivate文件夹

#建立退出环境下的文件夹
mkdir -p /home/username/anaconda3/envs/pytorch/etc/conda/deactivate.d
#写入脚本
vi /home/username/anaconda3/envs/pytorch/etc/conda/deactivate.d/deactivate.sh

添加如下内容：

export CUDA_HOME=$ORIGINAL_CUDA_HOME
export LD_LIBRARY_PATH=$ORIGINAL_LD_LIBRARY_PATH
unset ORIGINAL_CUDA_HOME
unset ORIGINAL_LD_LIBRARY_PATH

4.测试

测试

首先先在本机查看环境变量 CUDA_HOME

echo $CUDA_HOME
结果为/usr/local/cuda-10.2

激活环境后

echo $CUDA_HOME
结果为/usr/local/cuda-11.2