Ubuntu 16.04 配置 GPU驱动 CUDA cuDNN

参考文章

• ##准备工作
  检查GPU是否支持CUDA
  lspci | grep -i nvidia
  验证自己的Linux版本是否支持 CUDA
  $ uname -m && cat /etc/*release
  验证是否安装了gcc
  $ gcc --version
  验证系统是否安装了kernel header和 package development
  uname -r
  安装NVIDIA驱动程序(如果事先有装但失败的，先卸载)
  $ sudo apt remove --purge nvidia*
  禁用nouveau驱动
  lsmod | grep nouveau

  在Ubuntu16.04中，我们可以通过以下操作禁用nouveau
  a.在/etc/modprobe.d中创建文件blacklist-nouveau.conf
  $ sudo vim /etc/modprobe.d/blacklist-nouveau.conf
  在文件中输入以下内容：
  blacklist nouveau
  options nouveau modeset=0
  b.执行
  $ sudo update-initramfs -u
  c.查看nouveau是否已经成功禁用
  $ lsmod | grep nouveau
  关闭图形化界面
  $ sudo service lightdm stop

• apt 安装 显卡驱动(最好禁用 nouveau)

1. 添加Graphic Drivers PPA

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

2. 寻找合适的驱动版本

   $ ubuntu-drivers devices
   

3. 安装重启

   $ sudo apt install nvidia-
   $ sudo nvidia-smi
   $ sudo nvidia-settings
   

• 安装CUDA,去官网下载

1.安装

sudo sh cuda_9.1.85_387.26_linux.run

2. 安装一些库

 sudo apt install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

3. 设置环境变量

$ export PATH=/usr/local/cuda-9.1/bin:$PATH
$ export LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64:$LD_LIBRARY_PATH
$ export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.1

4. 测试

$ cd /home/user_name/NVIDIA_CUDA-9.1_Samples
$ make
$ cd bin/x86_64/linux/release
$ ./deviceQuery  

5. 卸载cuda
   To remove CUDA Toolkit:

$ sudo apt-get --purge remove "*cublas*" "cuda*"

To remove NVIDIA Drivers:

$ sudo apt-get --purge remove "*nvidia*"

Use the following command to uninstall a Toolkit runfile installation:

$ sudo /usr/local/cuda-X.Y/bin/uninstall_cuda_X.Y.pl

• 安装CUDNN 去官网下载

$ tar -xzvf cudnn-9.1-linux-x64-v7.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*

• 一些版本对应关系

  1. CUDA与Driver的对应版本
     https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
  2. cuda8.0与Linux系统以及GCC的对应关系
     https://docs.nvidia.com/cuda/archive/8.0/cuda-installation-guide-linux/index.html
  3. cuda9.0与Linux系统以及GCC的对应关系
     https://docs.nvidia.com/cuda/archive/9.0/cuda-installation-guide-linux/index.html
  4. cuda10.0与Linux系统以及GCC的对应关系
     https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
  5. Tensorflow cuda cudnn
     https://tensorflow.google.cn/install/source