1. 程式人生 > >ubantu16.04下原始碼編譯安裝tensorflow

ubantu16.04下原始碼編譯安裝tensorflow

Ubuntu16.04原始碼編譯安裝CPU版本的tensorflow

安裝說明:本教程針對tensorflow原始碼編譯安裝cpu版本,參考於tensorflow官網

安裝環境說明:

os:Ubuntu 16.04

arch:x86_64

tensorflow版本:1.0.1

python版本:python2

tensorflow的一些知識可以訪問官網

$ git clone https://github.com/tensorflow/tensorflow 
下載tensorflow的原始碼包
2.輸入命令:
$ cd tensorflow
$ git checkout Branch # where Branch
is the desired branch
$ git checkout r1.0
說明:這裡r1.0代表的是tensorflow的版本,讀者可以根據自己需要進行更改。
3.接下來準備好linux下的編譯環境
-Bazel
-python tensorflow 依賴
(1)安裝bazel
依次輸入命令:$ sudo apt-get install openjdk-8-jdk
         $ echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
         $ curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -
      $ sudo apt-get update && sudo apt-get install bazel
(2)安裝tensorflow依賴
輸入命令:$ sudo apt-get install python-numpy python-dev python-pip python-wheel
(如果使用的是Python3:則改為 $ sudo apt-get install python3-numpy python3-dev python3-pip python3-wheel)
(3)安裝支援tensorflow GPU的nvidia套件(具體見nvidia官網)
注:電腦gpu的計算能力必須大於或等於3.0。GPU的計算能力表可以取nvidia官網查
(4)輸入命令:$ sudo apt-get install libcupti-dev 
4.到這裡環境已經配置好了,接下來就可以原始碼編譯安裝了。
參考網站:https://www.tensorflow.org/install/install_sources#ConfigureInstallation(需翻牆)
5.$ bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package 
6.$ sudo pip install /tmp/tensorflow_pkg/tensorflow-1.0.1-py2-none-any.whl(注:這裡需根據自己使用的版本進行修改)
7.如果上面沒出問題的話,到這裡基本就大功告成了。開啟python,輸入import tensorflow,如果不報錯,就說明可以正常使用了!