Tensorflow GPU版 安裝教程

　　前一段時間按照官網給出的教程安裝了一遍，最近重裝又要重把運行環境重新安裝一次，又折騰了好久，主要是Tensorflow官網打開太慢，僅此作為記錄，以便以后使用。

參考資料：https://www.tensorflow.org/install/install_linux

 

  運行環境與安裝包：

　　Ubuntu 16.04，Python 2.7，Virtualenv， Cuda8.0，Cudnn V5 for Cuda 8.0，NVIDIA顯卡評分>3.0(GTX1060)

  第一步 安裝Virtualenv、tensorflow：

1.安裝 pip 與 virtualenv 

 $ sudo apt-get install python-pip python-dev python-virtualenv 

2.創建獨立的python運行環境

 $ virtualenv --system-site-packages tensorflow

3.激活tensorflow運行環境

 $ source ~/tensorflow/bin/activate # bash, sh, ksh, or zsh

4. 安裝tensorflow

(tensorflow)$ pip install --upgrade tensorflow-gpu

第二步：安裝cuda與cudnn

下載地址：https://developer.nvidia.com/cuda-downloads

下載CUDA

安裝CUDA：

1.  $ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
2.  $ sudo apt-get update
3.  $ sudo apt-get install cuda

設置環境變量LD_LIBRARY_PATH與CUDA_HOME：

打開配置文件

vim  ~/.bash_rc

在文件末尾添加下面兩行

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"

export CUDA_HOME=/usr/local/cuda

安裝Cudnn：

下載地址：https://developer.nvidia.com/cudnn

需要注冊，並填寫一堆表格。。。

解壓 cudnn，並copy到/usr/local/cuda文件夾

1. $  tar xvzf cudnn-8.0-linux-x64-v5.1.tgz

2. $ sudo cp cuda/include/cudnn.h /usr/local/cuda/include

3. $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

4. $ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* #給所有用戶添加讀的權限

安裝libcupti-dev庫：

$ sudo apt-get install libcupti-dev

NVIDIA顯卡驅動：

 

編輯日期：2017-06-04 10:51:47