使⽤用指南
安装 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 ,关闭该环境时
再移除.
- 创建环境启动时的脚本:
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即可)
- 创建环境关闭时的脚本:
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