ubuntu16.04安装cuDNN的两种方式以及验证

时间:2024-05-21 20:45:42

一、Tar File形式的安装

Tar File的下载如下图所示,选择红方框中的选项进行下载

ubuntu16.04安装cuDNN的两种方式以及验证

下载的是cudnn-*tgz的压缩包时,按下方指令进行安装:

首先解压缩下的cudnn压缩包文件

tar -xzvf cudnn-9.0-linux-x64-v7.tgz

执行安装,其实就是拷贝头文件和库文件并给予权限

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 /usr/local/cuda/lib64/libcudnn*

二、Debian File形式的安装

Debian File的下载如下图所示,选择红方框中的选项进行下载,此时可以选择下载三个

ubuntu16.04安装cuDNN的两种方式以及验证

 此时执行以下三个命令进行安装

sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb

三、验证cudnn是否安装成功

当选择Debian File进行安装时会在/usr/src/cudnn_samples_v7有一些cudnn的例子

编译mnistCUDNN sample进行验证

cp -r /usr/src/cudnn_samples_v7/ $HOME
cd  $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN

如果安装成功将会有如下图所示的:Test passed!

ubuntu16.04安装cuDNN的两种方式以及验证