Ubuntu16.04安装tensorflow_gpu教程

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

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

配置是ubuntu16.04, cuda8.0, cudnn5.1, tensorflow-gpu

另一种安装的是ubuntu18.04, cuda9.0, cudnn7, tensorflow-gpu.

这里是ubuntu16.04

 

一，安装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，根据下载的文件名(注意cuda版本与ubuntu版本有关的)相应的输入安装命令，我的文件名是：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}}

    4）重启电脑，再输入：nvidia-smi

           显示如下，则成功安装cuda

四，安装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

安装成功

六，测试

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

再输入：import tensorflow as tf

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