linux 安裝tensorflow（gpu版本）

一、安裝cuda

具體安裝過程見我的另一篇博客，ubuntu16.04下安裝配置深度學習環境

二、安裝tensorflow

1.具體安裝過程官網其實寫的比較詳細，總結一下的話可以分為兩種：安裝release版本和源碼編譯安裝。因為源碼編譯安裝比較繁瑣，且需要安裝谷歌自己的編譯器bazel，所以我選擇安裝編譯好的。

2.我寫這篇博客的時候tensorflow更新到了1.4.0，安裝編譯好的一定看版本，因為每個版本依賴的底層庫是不一樣的。

1.4.0版本安裝之前需要安裝CUDA-8，cuDNN v6.0.，ibcupti-dev library

注意以上軟件的版本，cuda一定要是8，cudnn得是6.0（我之前裝的是1.2.0，則cudnn是v5.1就行，所以這個版本很重要，如果你的cudnn版本過低的話會報錯：ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory）

安裝bcupti-dev library可執行

sudo apt-get install libcupti-dev

 

3.以上環境准備好后，安裝是很簡單的

如果你用的是anaconda，安裝步驟如下：

conda create -n tensorflow python=2.7 # or python=3.3, etc.

source activate tensorflow

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

如果是python直接安裝的話：
首先需要pip，如果沒有的話可以使用下面的命令安裝

sudo apt-get install python-pip python-dev   # for Python 2.7
sudo apt-get install python3-pip python3-dev # for Python 3.n

下面使用pip安裝tensorflow

$ pip install tensorflow # Python 2.7; CPU support (no GPU support)

$ pip3 install tensorflow # Python 3.n; CPU support (no GPU support)

$ pip install tensorflow-gpu # Python 2.7; GPU support

$ pip3 install tensorflow-gpu # Python 3.n; GPU support

不加版本的話默認是最新版本，如果想下特定的版本可在tensorflow后面加上版本號，例如第一個可以寫成pip install tensorflow==1.1.0

三、測試tensorflow是否安裝成功

經過一二兩個步驟，應該是安裝成功了，可寫個tensorflow的小程序測試一下

# Python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))