使?用指南
安裝 Anaconda3
檢查是否已經(jīng)安裝anaconda，看 which conda 是否有輸出 若沒(méi)有，按照 https://docs.anaconda.com/anaconda/install/進(jìn)?行行安裝
創(chuàng)建anaonda環(huán)境
比如現(xiàn)在要使?用tensorflow1.13，它要求?用cuda 10.0.
第?步 創(chuàng)建環(huán)境
conda create -n tf113_cuda10 python=3.6
環(huán)境名為 tf113_cuda10 ，指定python版本為 3.6 (tensorflow尚不不?支持3.7及以上的版本)
第?步 指定該環(huán)境使?用的cuda版本
tensorflow運(yùn)?行行時(shí)，實(shí)際是從動(dòng)態(tài)鏈接庫(kù)的地址找需要的cuda動(dòng)態(tài)鏈接庫(kù)，所以我們只需要指定 LD_LIBRARY_PATH . 即保證該環(huán)境啟動(dòng)時(shí)，把cuda 10的地址加?入 LD_LIBRARY_PATH ，關(guān)閉該環(huán)境時(shí)
再移除.
- 創(chuàng)建環(huán)境啟動(dòng)時(shí)的腳本:
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，把上?面內(nèi)容中的10.0換成9.0即可)
- 創(chuàng)建環(huán)境關(guān)閉時(shí)的腳本:
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
第三步 使?，檢查
啟動(dòng)環(huán)境: source activate tf113_cuda10 檢查動(dòng)態(tài)鏈接庫(kù)路路徑:
echo $LD_LIBRARY_PATH ，應(yīng)該看到輸出:
/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:
關(guān)閉環(huán)境: source deactivate
