Ubuntu16.04安装tensorflow_gpu教程

前言:之前安装了很多次,不分gpu/cpu版胡乱装,导致后面出现各种问题。不同于安装tensorflow_cpu版

gpu版基本按照点击打开链接安装,但是有些问题,下面具体说明。

一,安装pip

命令行输入:sudo apt-get install python-pip python-dev

可能会出现错误,如无法定位软件包 python-pip  软件包python-dev 没有可安装候选,这时候需要更新下软件包信息,

具体做法是在terminal终端下执行 sudo apt-get update

再执行上一句

二,安装nvidia driver

驱动版本推荐:ubuntu-drivers devices

我的输出:

vendor   : NVIDIA Corporation

model    : GM107M [GeForce GTX 960M]

modalias : pci:v000010DEd0000139Bsv000017AAsd00003802bc03sc02i00

driver   : xserver-xorg-video-nouveau - distro free builtin

driver   : nvidia-384 - distro non-free recommended

所以安装命令:sudo apt-get install nvidia-384

三,安装cuda 8.0

(1) 注意作者给的不是8.0版,下载8.0请移步点击打开链接, 或者百度云:点击打开链接

(2) 找到下载目录,在下载目录打开terminal,根据下载的文件名相应的输入安装命令,我的文件名是:8.0.61_375.26_linux.run

所以安装命令为:

sudo sh cuda_8.0.61_375.26_linux.run --override

(3) 配置cuda环境变量方式错误,按作者方式,将导致笔记本重启后,cuda安装失败。正确方式如下:

    1)打开bashrc文件:sudo gedit ~/.bashrc

    2) 在文件最后输入:

  1. export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}

    3)其次在bashrc文件最后添加一行依赖库:

  1. export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

Ubuntu16.04安装tensorflow_gpu教程

    4)重启电脑,再输入:nvidia-smi

           显示如下,则成功安装cuda

Ubuntu16.04安装tensorflow_gpu教程

四,安装cudnn 5.1

(1) 作者给的同样不是5.1版,下载5.1请移步点击打开链接,需要注册,或者百度云:点击打开链接

安装过程就是解压:

tar xvzf cudnn-8.0-linux-x64-v5.1.tgz

然后里面的文件拷贝到cuda文件夹下:

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*

五,安装tensorflow-gpu版

1. 可以直接安装,输入:sudo pip install tensorflow-gpu==1.0

2. 也可以按照作者的方式,不过他的不完整,且版本过低。

作者给的是不完整的命令,正确的命令是:(1)export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

(2)sudo pip install --upgrade $TF_BINARY_URL

可能也会出错,试试多输入几遍,我试了四遍成功了。

Successfully installed funcsigs-1.0.2 mock-2.0.0 numpy-1.14.1 pbr-3.1.1 protobuf-3.0.0 setuptools-38.5.2 six-1.11.0 tensorflow-0.11.0 wheel-0.30.0

安装成功

Ubuntu16.04安装tensorflow_gpu教程

六,测试

测试下,输入终端输入: python

再输入:import tensorflow as tf

显示如下,即成功安装tensorflow-gpu

Ubuntu16.04安装tensorflow_gpu教程