1. 查看配置好的CUDA和cuDNN版本,在我之前博客有提到,CUDA 8.0 和cuDNN 6.0, 查看Tensorflow對應版本:
2. Anaconda3 安裝
去清華鏡像源https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/下載Anaconda 3-4.2.0版本
命令:
cd Download
bash Anaconda3-4.2.0-Linux-x86_64.sh
修改環境:
cd ~/.bashrc
沒有就添加
export PATH="/用戶名/anaconda3/bin:$PATH"
source ~/.bashrc
驗證:
python
3. 通過Anaconda 安裝Tensorflow
創建新的conda環境,激活並安裝TensorFlow_gpu 1.4.0版本
命令:
conda create -n tensorflow-gpu python=3.5
source activate tensorflow-gpu
pip intall tensorflow-gpu==1.4.0
source deactivate tensorflow-gpu
驗證:
import tensorflow as tf