參考鏈接：
給虛擬環境指定cuda
本文重述原作者內容，主要是防止原鏈接丟失。
服務器上裝了多個版本的cuda，有的時候需要使用TensorFlow，有的時候需要使用pytorch，有時候需要使用cuda9.0+，有的時候需要使用cuda10.0+。那麼必然涉及到給每個不同的虛擬環境配上不同的cuda版本。
本文以有cuda9.0 和 cuda10.1爲例，介紹如何給pytorch這個環境指定cuda10.1版本，啓動環境時自動執行腳本激活cuda10.1。
1.首先激活虛擬環境 pytorch，輸出當前環境的路徑
conda activate pytorch
echo ${CONDA_PREFIX}
得到路徑, for example：/home/username/anaconda3/envs/pytorch
2.建立激活環境下的文件夾，寫入腳本
#建立激活環境下的文件夾
mkdir -p /home/username/anaconda3/envs/pytorch/etc/conda/activate.d
#寫入腳本
vi /home/username/anaconda3/envs/pytorch/etc/conda/activate.d/activate.sh
寫入內容如下：
3.建立退出環境下的文件夾，寫入腳本
#建立退出環境下的文件夾
mkdir -p /home/username/anaconda3/envs/pytorch/etc/conda/deactivate.d
#寫入腳本
vi /home/username/anaconda3/envs/pytorch/etc/conda/deactivate.d/deactivate.sh
寫入內容如下：
4.測試
首先先在本機查看環境變量 CUDA_HOME
echo $CUDA_HOME
結果爲/usr/local/cuda-9.0
激活環境後
echo $CUDA_HOME
結果爲/usr/local/cuda-10.1
注：如果出現從虛擬環境退出後，echo $CUDA_HOME 結果爲空值的情況，操作如下
執行初始化文件，立即生效
source .bashrc
結束。