Ubuntu 16.04 安装 Tensorflow(GPU支持)

前端之家收集整理的这篇文章主要介绍了Ubuntu 16.04 安装 Tensorflow(GPU支持)前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

原文链接

http://blog.topspeedsnail.com/archives/tag/tensorflow

本文记录Ubuntu 16.04安装Tensorflow步骤,也包括怎么从源码编译安装Tensorflow。

要想安装Tensorflow GPU版本,你需要有一个新一点的Nvidia显卡。

Tensorflow cpu版本的安装

$ sudo apt-get install python-pip python-dev python-virtualenv    # python 2.7
$ sudo apt-get install python3-pip python3-dev python3-virtualenv # python 3.4+


使用虚拟环境(可选):Python虚拟环境(pyvenv、virtualenv)

$ virtualenv --system-site-packages ~/tensorflow
$ source ~/tensorflow/bin/activate
# Ubuntu/Linux 64-bit,cpu only,Python 2.7
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp27-none-linux_x86_64.whl
 
# Ubuntu/Linux 64-bit,Python 3.4
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp34-cp34m-linux_x86_64.whl
 
# Ubuntu/Linux 64-bit,Python 3.5
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp35-cp35m-linux_x86_64.whl

安装Tensorflow:

# Python 2
$ sudo pip install --upgrade $TF_BINARY_URL
 
# Python 3
$ sudo pip3 install --upgrade $TF_BINARY_URL


如果要升级Tensorflow,替换新版本的TF_BINARY_URL。https://www.tensorflow.org

编译安装Tensorflow(GPU支持

安装NVidia显卡驱动,你可以在Ubuntu内置的附加驱动中安装。

$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt update

安装CUDA:

#下载安装:
#https://developer.nvidia.com/cuda-toolkit
$ sudo sh cuda_8.0.44_linux.run --override   # 安装位置: /usr/local/cuda
 
# 默认仓库中的版本较旧
#$ sudo apt install nvidia-cuda-toolkit nvidia-cuda-dev   # 安装位置: /usr

安装CudNN V5:https://developer.nvidia.com/cudnn

# 下载CudNN 5.1 for Cuda 8.0
$ sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*