下載Tensorflow源碼
git clone https://github.com/tensorflow/tensorflow
如果無法下載也可以在github上直接下載tensorflow的打包文件,這樣也能編譯,但是不能使用git命令
可根據需要切換到不同的分支
安裝bazel
輸入以下命令
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
如果無法下載,也可以自己下載文件包,然后放入tensorflow目錄中
安裝依賴庫
# Python 2.7
sudo apt-get install python-numpy python-dev python-pip python-wheel
# Python 3.x
sudo apt-get install python3-numpy python3-dev python3-pip python3-wheel
安裝CUDA和cuDNN
如果只使用CPU編譯可以跳過此步驟
編譯安裝
cd tensorflow
./configure
此時會逐項彈出選項,可以根據自己的實際需求選擇需要啟用的功能
編譯CPU版本(//后面也是命令,不是注釋)
bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package
編譯GPU 版本
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
GCC 5需要設置--cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0"選項。
生成python wheel包,其中/tmp/tensorflow_pkg是你想安裝tf的目錄
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
最后安裝生成的包
sudo pip install /tmp/tensorflow_pkg/tensorflow-1.2.0-py2-none-any.whl