【TensorFLow】Ubuntu16.04安装CUDA8.0+cuDNN6.0+TensorFlow

NVIDIA驱动

---

Ubuntu在安装NVIDIA驱动前要先将BIOS中的的Secure Boot设置为Disabled

附加驱动安装

最简单的安装方式是直接在“系统设置->软件和更新->附加驱动”中安装

终端安装

如果需要在终端中安装，可以按照以下步骤进行

1. 删除旧NVIDIA驱动

sudo apt-get purge nvidia*

2. 禁用nouveau驱动

打开黑名单文件blacklist.conf

sudo vim /etc/modprobe.d/blacklist.conf

在末尾添加

blacklist vga16fb 
blacklist nouveau 
blacklist rivafb 
blacklist rivatv 
blacklist nvidiafb

然后更新内核

sudo update-initramfs -u

重启系统确认nouveau是否已经被屏蔽掉

lsmod | grep nouveau

3. 安装NVIDIA驱动

添加Graphic Drivers PPA

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

寻找合适的驱动版本

ubuntu-drivers devices

按ctrl+alt+F1进入tty1模式，关闭(图形)桌面显示管理器LightDM

sudo service lightdm stop

安装nvidia driver，安装完成后重启

sudo apt-get install nvidia-384
sudo reboot

重启系统后，执行下面的命令查看驱动的安装状态显示安装成功

sudo nvidia-smi

CUDA

---

首先在官网下载CUDA8.0

我下载的是deb包，所以直接安装就行了

$ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
$ sudo apt-get install cuda

如果是run文件，则运行即可，中间过程中注意不要选择安装NVIDIA驱动，其它都可以选择yes

sudo sh cuda_8.0.44_linux.run

然后打开配置文件profile

$ sudo vim /etc/profile

在末尾添加以下环境变量

# setting the environment variables so CUDA will be found
# After open profile, Add follow code at the end of file
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

然后更新环境变量

$ . /etc/profile

cuDNN

---

首先在官网下载cuDNN6.0，需要登录填写一个调查问卷

然后进入下载目录进行解压

tar zxvf  cudnn-8.0-linux-x64-v6.0.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

---

安装方法可以参考极客学院-下载与安装

我用Python3安装的TensorFlow

sudo pip3 install -U --pre setuptools wheel
sudo pip3 install -U --pre numpy scipy matplotlib scikit-learn scikit-image
sudo pip3 install -U --pre tensorflow-gpu

想要加快速度的话，请参照这篇博客这里使用国内镜像进行安装。

安装完毕后，在终端中输入python（或python3），然后输入：

import tensorflow

没有报错说明你已经可以正常使用了