本文記錄Ubuntu 16.04安裝Tensorflow步驟，也包括怎么從源碼編譯安裝Tensorflow。

要想安裝Tensorflow GPU版本，你需要有一個新一點的Nvidia顯卡。

Tensorflow CPU版本的安裝

1

2

$sudoapt-getinstall-pippython-devpython-virtualenv# python 2.7

$sudoapt-getinstallpython3-pippython3-devpython3-virtualenv# python 3.4+

1

2

$virtualenv--system-site-packages~/tensorflow

$source~/tensorflow/bin/activate

1

2

3

4

5

6

7

8

# Ubuntu/Linux 64-bit, CPU only, Python 2.7

$exportTF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp27-none-linux_x86_64.whl

# Ubuntu/Linux 64-bit, CPU only, Python 3.4

$exportTF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp34-cp34m-linux_x86_64.whl

# Ubuntu/Linux 64-bit, CPU only, Python 3.5

$exportTF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp35-cp35m-linux_x86_64.whl

安裝Tensorflow：

1

2

3

4

5

# Python 2

$sudopipinstall--upgrade$TF_BINARY_URL

# Python 3

$sudopip3install--upgrade$TF_BINARY_URL

如果要升級Tensorflow，替換新版本的TF_BINARY_URL。https://www.tensorflow.org

編譯安裝Tensorflow(GPU支持)

安裝NVidia顯卡驅動，你可以在Ubuntu內置的附加驅動中安裝。

1

2

$sudoadd-apt-repositoryppa:graphics-drivers/ppa

$sudoaptupdate

安裝CUDA：

1

2

3

4

5

6

#下載安裝：

#https://developer.nvidia.com/cuda-toolkit

$sudoshcuda_8.0.44_linux.run--override# 安裝位置: /usr/local/cuda

# 默認倉庫中的版本較舊

#$ sudo apt install nvidia-cuda-toolkit nvidia-cuda-dev? # 安裝位置: /usr

1

2

3

4

5

# 下載CudNN?5.1 for Cuda 8.0

$sudotar-xzvfcudnn-8.0-linux-x64-v5.1.tgz

$sudocpcuda/include/cudnn.h/usr/local/cuda/include

$sudocpcuda/lib64/libcudnn*/usr/local/cuda/lib64

$sudochmoda+r/usr/local/cuda/include/cudnn.h/usr/local/cuda/lib64/libcudnn*

在~/.bashrc文件中添加環境變量：

1

2

exportLD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"

exportCUDA_HOME=/usr/local/cuda

使環境變量生效：

1

$source~/.bashrc

下載tensorflow源碼：

1

2

$cd~

$gitclonehttps://github.com/tensorflow/tensorflow

安裝一些編譯和依賴工具：

1

$sudoapt-getinstalldefault-jdkpython-devpython3-devpython-numpypython3-numpybuild-essentialpython-pippython3-pippython-virtualenvswigpython-wheellibcurl3-dev

安裝Bazel：

1

2

3

4

5

$echo"deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8"|sudotee/etc/apt/sources.list.d/bazel.list

$curlhttps://storage.googleapis.com/bazel-apt/doc/apt-key.pub.gpg|sudoapt-keyadd-

$sudoapt-getupdate

$sudoapt-getinstallbazel

$sudoapt-getupgradebazel

配置編譯選項：

1

2

$cd~/tensorflow

$./configure

需要輸入Python路徑，默認是?/usr/bin/python。如果你使用Python3，輸入：/usr/bin/python3.5。

輸入Python模塊路徑，默認是/usr/local/lib/python2.7/dist-packages。如果你使用Python3，輸入：/usr/local/lib/python3.5/dist-packages。

輸入Cuda SDK版本和Cudnn版本：8.0、5.1.5。

配置完成，輸入如下信息：

1

2

INFO:Allexternaldependenciesfetchedsuccessfully.

Configurationfinished

編譯tensorflow：

1

2

3

$bazelbuild-copt--config=cuda# GPU支持

# CPU支持

#$ bazel build -c opt

構建pip包：

1

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

安裝pip包：

1

2

$sudopipinstall/tmp/tensorflow_pkg/tensorflow# python2

$sudopip3install/tmp/tensorflow_pkg/tensorflow# python3

參考：

https://www.tensorflow.org/versions/r0.11/get_started/os_setup.html#download-and-setup

總結

以上是生活随笔為你收集整理的linux安装tensorflow教程,Ubuntu 16.04 安装 TensorFlow(GPU支持)的全部內容，希望文章能夠幫你解決所遇到的問題。

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