Ubuntu 16.04下安裝Tensorflow(GPU)

參考：http://blog.sina.com.cn/s/blog_672f698e0102wavp.html

1.首先安裝nvidia顯卡驅動:

系統設置->軟件更新->附加驅動->選擇nvidia最新驅動(361)->應用更改

​2.下載CUDA8.0 地址https://developer.nvidia.com/cuda-release-candidate-download(需要登陸)

3.安裝cuda 

sudo dpkg -i cuda-repo-ubuntu1604-8-0-rc_8.0.27-1_amd64​.deb

sudo apt-get update

sudo apt-get install cuda​

4.下載cudnn,最好是v4版本 地址https:developer.nvidia.com/cudnn

5.把cudnn解壓復制到cuda下

tar xvzf cudnn-7.0-linux-x64-v4.0-prod.tgz

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

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*​​

6.從源碼安裝tensorflow

 

3、ubuntu的gcc編譯器是5.4.0，然而cuda8.0不支持5.0以上的編譯器，因此需要降級，把編譯器版本降到4.9：

[python] view plain copy

sudo apt-get install g++-4.9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10
sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30
sudo update-alternatives --set cc /usr/bin/gcc
sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
sudo update-alternatives --set c++ /usr/bin/g++

 

git clone https://github.com/tensorflow/tensorflow​

進入tensorflow根目錄(home下)運行./configure配置tensorflow

7.安裝Bazel（最好下好了再安裝，第二種方法，不然需要翻牆）

http://www.bazel.io/docs/install.html​

安裝其他依賴

sudo apt-get install python-numpy swig python-dev python-wheel​​

8.建立GPU支持

bazel build -c opt --config=cuda //tensorflow/cc:tutorials_example_trainer​ 如發生錯誤可在后面加上--verbose_failures運行如下

bazel build -c opt --config=cuda //tensorflow/cc:tutorials_example_trainer​ --verbose_failures

完成后運行

​bazel-bin/tensorflow/cc/tutorials_example_trainer --use_gpu​

此程序會計算一個2*2矩陣的主特征值 輸出如下

 

創建PIP安裝包

bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package​

--config=cuda表示支持gpu

bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg​

sudo pip install /tmp/tensorflow_pkg/tensorflow-0.9.0-py2-none-any.whl

OK,搞完收工​​