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即可)

  1. 创建环境关闭时的脚本:
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

你可能感兴趣的:(Anaconda中使用不同版本的cuda)