下載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

總結

以上是生活随笔為你收集整理的cuda 编译 linux,Linux下安装Tensorflow源码及编译的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。