conda使用不同版本cuda

利用脚本在进入环境的时候切换环境变量CUDA_HOME到需要的版本，退出环境后切换为原CUDA版本
1.建立激活环境下的文件夹，写入脚本

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

写入

ORIGINAL_CUDA_HOME=$CUDA_HOME
ORIGINAL_LD_LIBRARY_PATH=$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-版本号
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

2.建立退出环境下的文件夹，写入脚本

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

vim写入

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

执行

source ~/.bashrc

重连服务器，激活环境后输入

echo $CUDA_HOME

可以得到

/usr/local/cuda-8.0

由原来的11切换为了8```