Ubuntu 16.04 安裝CPU版tensorflow

1.下載tensorflow源碼:

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

2.安裝JDK

$ sudo apt-get install openjdk-8-jdk

3.安裝bazel(apt):

$ 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

安裝bazel(binary installer):

$ sudo apt-get install pkg-config zip g++zlib1g-dev unzip python

Download bazel installer: https://github.com/bazelbuild/bazel/releases

$ chmod +x bazel--installer-linux-x86_64.sh

$ ./bazel--installer-linux-x86_64.sh --user

$ vim ~/.bashrc

添加:export PATH="$PATH:$HOME/bin"

4.安裝依賴庫:

# for Python 2.7

$ sudo apt-get install python-numpypython-dev python-pip python-wheel

# for Python 3.x

$ sudo apt-get install python3-numpy python3-dev python3-pip python3-wheel

5.配置:

$ cd tensorflow*

$ ./configure

安裝命令行提示,逐個設置編譯選項(最簡單的版本一路默認,選no)

6.編譯:

$ bazel build --config=opt --incompatible_load_argument_is_label=false //tensorflow/tools/pip_package:build_pip_package

7.生成whl包:

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

8.安裝生成的包:

$ sudo pip install /tmp/tensorflow_pkg/tensorflow-*.whl

9.重啟:

$ sudo reboot


備注:

遇到“The?following packages have unmet dependencies:”的問題:

$ sudo apt-get clean

$ sudo apt-get autoclean

$ sudo apt-get? -f install

$ sudo dpkg --configure -a

$ sudo apt-get -f install


Bazel卸載

$ sudo apt-get --purge remove bazel

$ sudo apt autoremove

Tensorflow修改源代碼后重新編譯

#!/bin/bash

sudo pip uninstall tensorflow -y

bazel build --config=opt --incompatible_load_argument_is_label=false //tensorflow/tools/pip_package:build_pip_package

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

sudo pip install /tmp/tensorflow_pkg/tensorflow-*.whl

查看tensorflow版本和路徑信息:

python環境執行:

>>> import tensorflow as tf

>>> tf.__version__

輸出tensorflow版本信息

>>>tf.__path__

輸出tensorflow安裝路徑

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。