Ubuntu，NVIDIA GPU和Tensorflow-GPU的安装

一. 电脑硬件说明

• Ubuntu 16.06
• GeForce GTX 960M
  
• 驱动：384.130
• CUDA：9.0
• cuDNN：7.1.2

二. 安装NVIDIA GPU的驱动

1. 选择GPU驱动下载

首先查询自己电脑显卡的型号：

lspci | grep -i nvidia

在NVIDIA[https://www.nvidia.com/Download/index.aspx?lang=en-us]选择合适的Driver的下载，按照笔者的型号选择如下：

点击search后下载。

2. 安装GPU驱动

（1）. 屏蔽nouveau驱动

在/etc/modprobe.d/下新建一个文件blacklist-nouveau.conf，即：

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

写入：

blacklist nouveau  
options nouveau modeset=0  

更新

sudo update-initramfs -u  

测试：

lsmod | grep nouveau

如果没有任何输出则表明禁用成功

（2）. 安装GPU驱动

输入命令：

sudo service lightdm stop

进入驱动所在的文件夹：

sudo chmod +x ./NVIDIA-Linux-x86_64-384.130.run

修改权限。然后：

sudo ./NVIDIA-Linux-x86_64-384.111.run

安装完成后重启电脑：

reboot

重启后输入：

nvidia-smi

如果有打印GPU的相关信息则表示安装成功：

3. 安装CUDA

（1） 下载CUDA

在官网上选择合适的CUDA并下载runfile文件：

注意下载Base下面的四个补丁：

下载的时候注意版本的对应：

笔者下载目录为：