1. 安裝python

python環(huán)境使用anaconda

從官方網(wǎng)站下載操作系統(tǒng)對(duì)應(yīng)的版本

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

./Anaconda3-5.0.1-Linux-x86_64.sh

2. Nvidia顯卡驅(qū)動(dòng)

重啟計(jì)算機(jī)

3. Nvidia CUDA 8.0

從官方網(wǎng)站下載對(duì)應(yīng)的版本

sudo sh cuda_8.0.61_375.26_linux.run

注意：這是選擇不安裝顯卡驅(qū)動(dòng)，因?yàn)榍懊嬉寻惭b 。

安裝cuda 8.0 patch

sudo sh cuda_8.0.61.2_linux.run

參照界面提示配置環(huán)境變量/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

從官方網(wǎng)站下載cuDNN 6.0， 注意這一步需要注冊(cè)賬戶

下載完成后解壓縮。

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. 開發(fā)工具PyCharm

安裝過程略

7. 測(cè)試

--finished--

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。