Ubuntu16.04+CUDA9.0+CUDNN7.3.1.20环境安装

下载资源

• 1.CUDA9.0
• 2.CUDNN7.3.1

提醒

安装cuda时会同时安装NVIDIA显卡驱动

安装

• 安装cuda

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

• 重启

sudo reboot

• 配置环境

sudo nano /etc/bash.bashrc

在最后添加如下

export CUDA_HOME=/usr/local/cuda-9.0 
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} 
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

sudo nano  ~/.bashrc

在最后添加如下

export CUDA_HOME=/usr/local/cuda-9.0 
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} 
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

• 重启后测试

nvcc -V

• 安装cudnn
  将下载好的库文件解压出来后

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