【Ubuntu】给虚拟环境指定CUDA版本———笔记（配置FlowNet2）

【Ubuntu】给虚拟环境指定CUDA版本

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

 conda activate 环境名

输出当前环境路径

 echo ${CONDA_PREFIX}

2、activate文件夹

#建立激活环境下的文件夹

mkdir -p /home/username/anaconda3/envs/pytorch/etc/conda/activate.d

（mkdir -p 简单理解为创建目录，多个文件父类，比如 mkdir -p dir1/dir2/dir3
不会考虑文件夹是否存在，存在就创建子类，不存在就直接创建）

写入脚本，编辑上边创建的文件夹

	 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   ###修改cuda版本
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

vim 命令 保存并退出

:wq

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、测试

激活环境前后分别输入命令对比

echo $CUDA_HOME