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

安裝命令行提示,逐個設(shè)置編譯選項(最簡單的版本一路默認(rèn),選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環(huán)境執(zhí)行:

>>> import tensorflow as tf

>>> tf.__version__

輸出tensorflow版本信息

>>>tf.__path__

輸出tensorflow安裝路徑

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容