1. 程式人生 > >Ubuntu16.04安裝Tensorflow(GPU)簡明教程

Ubuntu16.04安裝Tensorflow(GPU)簡明教程

安裝tensorflow的方式有很多種,這裡介紹的是通過Bazel編譯tensorflow原始碼的方式來安裝tf。
ps:我使用的版本是r1.7

(1)安裝JDK8
新增PPA源

sudo add-apt-repository ppa:webupd8team/java
sudo apt-get update
sudo apt-get install openjdk-8-jdk

安裝完成後鍵入

java -version

若看到Java版本資訊則表明JDK安裝成功
這裡寫圖片描述
(2)安裝依賴包

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

(3)安裝Bazel

  • 方法一:

由於前面我們已經安裝了JDK8,所以這裡選擇bazel-<version>-without-jdk-installer-linux-x86_64.sh
其中<version>為Bazel的版本號,我使用版本是的是bazel-0.11.1-without-jdk-installer-linux-x86_64.sh
然後賦予其可執行許可權,執行:

sudo chmod +x bazel-0.11.1-without-jdk-installer-linux-x86_64.sh
.
/bazel-0.11.1-without-jdk-installer-linux-x86_64.sh --user

最後然後新增其路徑

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

重啟一下使其生效

  • 方法二

新增Bazel的PPA源,但需要科學上網。

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 upgrade bazel

(1)安裝一些依賴項

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

(2)下載Tensorflow
Github上的主頁:https://github.com/tensorflow/tensorflow
branch處可以選擇版本
解壓原始碼,進入解壓好的資料夾中,在當前目錄下開啟終端

./configure

配置安裝資訊

#這裡我用的是python3,python2的話預設即可
Please specify the location of python. [Default is /usr/bin/python]: /usr/bin/python3
Please input the desired Python library path to use.  Default is [/usr/local/lib/python3.5/dist-packages]
#除了CUDA以外,其他平臺都選n
Do you wish to build TensorFlow with CUDA support? [y/N]: y
#根據CUDA的版本選擇
Please specify the CUDA SDK version you want to use, e.g. 7.0. [Leave empty to default to CUDA 9.0]: 9.1
#根據cuDNN的版本選擇
Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 7.0]: 7.1
#使用GCC編譯器,這裡選n
Do you want to use clang as CUDA compiler? [y/N]: n
Do you wish to build TensorFlow with MPI support? [y/N]: n
Would you like to interactively configure ./WORKSPACE for Android builds? [y/N]: n
#這裡略去了一些選項

配置完成後會看到最後出現

Configuration finished

(3)通過pip安裝

bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg  
cd ///tmp/tensorflow_pkg
ls

通過ls命令確認安裝包名字,然後再通過pip安裝,<version>和安裝環境有關

sudo pip3 install /temp/tensorflow_pkg/tensorflow-<version>-none-any.whl

(4)測試
進入python,鍵入以下程式碼

import tensorflow as tf
hello = tf.constant('hello tensorflow')
sess = tf.Session()
print(sess.run(hello))

若看到如下輸出,則表明Tensorflow安裝成功

hello tensorflow

愉快的一天又開始了~~~