在需要更换cuda版本的环境中添加激活和退出时的配置文件,示例中环境为pytorch_0.4,更换为自己的环境名称,activate.sh中的cuda路径为想要在此环境中使用的cuda
# 创建文件
mkdir -p activate.d/activate.sh
sudo vim /home/kxc/.conda/envs/pytorch_0.4/etc/conda/activate.d/activate.sh
# 写入以下内容
ORIGINAL_CUDA_HOME=$CUDA_HOME
ORIGINAL_LD_LIBRARY_PATH=$LD_LIBRARY_PATH
ORIGINAL_LIBRARY_PATH=$LIBRARY_PATH
ORIGINAL_PATH=$PATH
export PATH=/us