Ubuntu16.04搭建深度学习环境

安装显卡驱动

1. 查看显卡型号

   lspci | grep -i vga
   

2. 下载显卡驱动

3. 卸载原系统中的Nvidia驱动。

   sudo apt-get remove --purge nvidia*
   

4. 禁用nouveau

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

   在最后添加代码段：

   blacklist vga16fb
   blacklist nouveau
   blacklist rivafb
   blacklist rivatv
   blacklist nvidiafb
   

   更新：

   sudo update-initramfs -u
   

   重新启动

5. 进入命令行模式，关闭图形界面：

   sudo service lightdm stop
   

6. 安装驱动

   sudo sh NVIDIA-Linux-x86_64-xxx.xx.run --no-opengl-files --no-x-check --no-nouveau-check
   

安装CUDA

7. 安装

   sudo ./cuda_xxx.run --no-opengl-libs
   

   在安装选项中可以只选择Cuda，不要选择Nvidia Driver.
8. 设置环境变量

   sudo gedit /etc/profile
   

   加入代码：

   export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
   export LD_LIBRARY_PATH=/usr/local/cuda/lib64\
   ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
   

   验证：

   nvcc -V
   

安装CUDNN

1. 解压cudnn包

   tar -zxvf cudnn-xxx.tgz
   

2. 拷贝文件

   sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
   sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d 
   sudo chmod a+r /usr/local/cuda/include/cudnn.h
   sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
   

3. 验证

   cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2