具體參考： https://zhuanlan.zhihu.com/p/137874101 （如果侵權(quán)获三，速刪）
Ubuntu安裝了多個版本的CUDA铺然，不同的虛擬環(huán)境需要切換不同的CUDA
本文以CUDA10.2和CUDA11.2為例：
1.首先激活虛擬環(huán)境 pytorch找蜜，輸出當(dāng)前環(huán)境的路徑
conda activate pytorch
echo ${CONDA_PREFIX}
得到路徑，例如：/home/username/anaconda3/envs/pytorch
2.activate文件夾
#建立激活環(huán)境下的文件夾
mkdir -p /home/username/anaconda3/envs/pytorch/etc/conda/activate.d
#寫入腳本
vi /home/username/anaconda3/envs/pytorch/etc/conda/activate.d/activate.sh
添加如下內(nèi)容：
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文件夾
#建立退出環(huán)境下的文件夾
mkdir -p /home/username/anaconda3/envs/pytorch/etc/conda/deactivate.d
#寫入腳本
vi /home/username/anaconda3/envs/pytorch/etc/conda/deactivate.d/deactivate.sh
添加如下內(nèi)容：
export CUDA_HOME=$ORIGINAL_CUDA_HOME
export LD_LIBRARY_PATH=$ORIGINAL_LD_LIBRARY_PATH
unset ORIGINAL_CUDA_HOME
unset ORIGINAL_LD_LIBRARY_PATH
4.測試
測試
首先先在本機(jī)查看環(huán)境變量 CUDA_HOME
echo $CUDA_HOME
結(jié)果為/usr/local/cuda-10.2
激活環(huán)境后
echo $CUDA_HOME
結(jié)果為/usr/local/cuda-11.2