Anaconda中使用不同版本的cuda

使⽤用指南

安装 Anaconda3

检查是否已经安装anaconda，看 which conda 是否有输出 若没有，按照 https://docs.anaconda.com/anaconda/install/进⾏行行安装

创建anaonda环境

比如现在要使⽤用tensorflow1.13，它要求⽤用cuda 10.0.

第⼀步 创建环境

conda create -n tf113_cuda10 python=3.6
环境名为 tf113_cuda10 ，指定python版本为 3.6 (tensorflow尚不不⽀支持3.7及以上的版本)

第⼆步 指定该环境使⽤用的cuda版本

tensorflow运⾏行行时，实际是从动态链接库的地址找需要的cuda动态链接库，所以我们只需要指定 LD_LIBRARY_PATH . 即保证该环境启动时，把cuda 10的地址加⼊入 LD_LIBRARY_PATH ，关闭该环境时
再移除.

1. 创建环境启动时的脚本:

mkdir -p ~/anaconda3/envs/tf113_cuda10/etc/conda/activate.d
touch ~/anaconda3/envs/tf113_cuda10/etc/conda/activate.d/activate.sh
vim ~/anaconda3/envs/tf113_cuda10/etc/conda/activate.d/activate.sh
chmod +x ~/anaconda3/envs/tf113_cuda10/etc/conda/activate.d/activate.sh

在脚本中填⼊入:

 #!/bin/sh
ORIGINAL_LD_LIBRARY_PATH=$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:/usr/local/cuda-
10.0/extras/CUPTI/lib64:/lib/nccl/cuda-10:$LD_LIBRARY_PATH

(若想⽤用cuda 9.0，把上⾯面内容中的10.0换成9.0即可)

2. 创建环境关闭时的脚本:

mkdir -p ~/anaconda3/envs/tf113_cuda10/etc/conda/deactivate.d
touch ~/anaconda3/envs/tf113_cuda10/etc/conda/deactivate.d/deactivate.sh
vim ~/anaconda3/envs/tf113_cuda10/etc/conda/deactivate.d/deactivate.sh
chmod +x
~/anaconda3/envs/tf113_cuda10/etc/conda/deactivate.d/deactivate.sh

在脚本中填⼊:

#!/bin/sh
export LD_LIBRARY_PATH=$ORIGINAL_LD_LIBRARY_PATH
unset ORIGINAL_LD_LIBRARY_PATH

第三步 使⽤，检查

启动环境: source activate tf113_cuda10 检查动态链接库路路径:
echo $LD_LIBRARY_PATH ，应该看到输出:

 /usr/local/cuda-10.0/lib64:/usr/local/cuda-
10.0/extras/CUPTI/lib64:/lib/nccl/cuda-10:

安装包: pip install tf-nightly-gpu (或者⽤用 conda install ) 检查安装包列列表: pip list 或者 conda list:
关闭环境: source deactivate