问题:在ubuntu16.04上安装cudnn,本机已经提前安装NVIDIA384和cuda9.0
参考:官方文档https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#install-linux
卸载之前安装的cudnn:
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
解决:
Linux安装有两种方式,一种下载tar包,一种下载deb包安装,这里推荐下载deb包安装,便于后面进行是否安装成功的验证。
cudnn文件的下载地址:https://developer.nvidia.com/rdp/cudnn-download,在这个页面下载下面三个文件
来到对应的下载目录,按照下面的命令进行安装:
sudo dpkg -i libcudnn7_7.3.1.20-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.3.1.20-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.3.1.20-1+cuda9.0_amd64.deb
执行完毕就安装成功了。
是否安装成功验证:
cp -r /usr/src/cudnn_samples_v7/ /home/
cd /home/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN
输出的最后如下即表示安装成功