1. 安裝python

python環境使用anaconda

從官方網站下載操作系統對應的版本

chmod +x Anaconda3-5.0.1-Linux-x86_64.sh

./Anaconda3-5.0.1-Linux-x86_64.sh

2. Nvidia顯卡驅動

重啟計算機

3. Nvidia CUDA 8.0

從官方網站下載對應的版本

sudo sh cuda_8.0.61_375.26_linux.run

注意：這是選擇不安裝顯卡驅動，因為前面已安裝 。

安裝cuda 8.0 patch

sudo sh cuda_8.0.61.2_linux.run

參照界面提示配置環境變量/etc/profile

export PATH=/usr/local/cuda-8.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:#LD_LIBRARY_PATH

4. Nvidia CuDNN 6.0

從官方網站下載cuDNN 6.0， 注意這一步需要注冊賬戶

下載完成后解壓縮。

tar zxvf cudnn-8.0-linux-x64-v6.0-tgz

cp cuda/include/* /usr/local/cuda-8.0/include/

cp cuda/lib64/* /usr/local/cuda-8.0/lib64/

5. Tensorflow GPU

sudo apt-get install libcupti-dev

conda create -n tensorflow python=3.6

source activate tensorflow

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.4.0-cp36-cp36m-linux_x86_64.whl

6. 開發工具PyCharm

安裝過程略

7. 測試

--finished--

總結

以上是生活随笔為你收集整理的Tensorflow GPU安装指南 (Ubuntu 16.04 anaconda cuda8.0 cuDNN6.0)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。