Ubuntu16.04 安装CUDA和CUDNN

1. 安装NVIDIA驱动

卸载掉原有驱动(如果是新配机器可以不用)
sudo apt-get remove –purge nvidia
安装驱动有两种办法

• 用ubuntu默认的附加驱动安装
• 在nvidia官网下载驱动安装

最近发现新官网下载的驱动有时问题比较多, 直接使用附加驱动nvidia-384就可以

系统默认驱动安装

进入附加驱动->选择nvidia-384->应用更改

安装成功后重启，检查nvidia-smi，有打印gpu信息的表格就说明成功了

安装cuda

• 下载
  到官网下载, 首选runfile
  需要注意驱动版本, 我的版本对应的是cuda9.0
• 安装
  安装前需要先关闭图形界面
  sudo service lightdm stop
  安装runfile文件
  注意: 安装时让选择是否安装驱动, 选择no

  chmod +x *.run
  sudo ./cuda_9.0.176_384.81_linux.run
  

  安装完成后打开图形界面
  sudo service lightdm restart

安装cudnn

• 下载
  到官网下载, 需要注册
• 安装
  解压cudnn-9.0-linux-x64-v7.tgz,
  tar -zxvf cudnn-9.0-linux-x64-v7.0.tgz
  复制解压好的文件到/usr/local/cuda下

  sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
  sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
  

添加环境变量

增加环境变量到~/.bashrc中

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

一切就绪重启计算机