ubuntu16.04安装配置CUDA8.0+cuDNN6.0

CUDA下载安装

下载地址,选择适合自己系统的版本

我这里是CUDA8.0，百度网盘下载地址，密码：9yf5

依次执行下列命令

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

至此，CUDA安装完成。

修改环境变量

使用下面命令打开bashrc文件

sudo gedit ~/.bashrc

在文件最后加上cuda配置

 

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

然后保存。

source ~/.bashrc

验证

nvcc -V

(watch -n 1) nvidia-smi

CUDA成功安装。

cuDNN配置

下载链接，同样是选择合适的版本

我的对应版本cudnn6.0，百度网盘下载地址，密码：3hov

解压，执行以下命令

tar -xzvf cudnn-8.0-linux-x64-v6.0.tgz

得到一个cuda文件夹，里面有两个文件夹，分别是include和lib64

执行以下命令，完成文件复制

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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

cuDNN配置完成。