在Ubuntu 18上安装cuDNN

最新推荐文章于 2024-10-14 23:04:42 发布

THMAIL

最新推荐文章于 2024-10-14 23:04:42 发布

阅读量991

点赞数 1

分类专栏：数据挖掘

本文链接：https://blog.youkuaiyun.com/THMAIL/article/details/85857571

版权

数据挖掘专栏收录该内容

5 篇文章

订阅专栏

INSTALL

Navigate to your <cudnnpath> directory containing the cuDNN Tar file.
Unzip the cuDNN package.
```
$ tar -xzvf cudnn-9.0-linux-x64-v7.tgz
```

Copy the following files into the CUDA Toolkit directory, and change the file permissions.

$ 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

Verifying

To verify that cuDNN is installed and is running properly, compile the mnistCUDNN sample located in the /usr/src/cudnn_samples_v7 directory in the debian file.

Copy the cuDNN sample to a writable path.
```
$cp -r /usr/src/cudnn_samples_v7/ $HOME
```
Go to the writable path.
```
$ cd  $HOME/cudnn_samples_v7/mnistCUDNN
```
Compile the mnistCUDNN sample.
```
$make clean && make
```
Run the mnistCUDNN sample.
```
$ ./mnistCUDNN
```
If cuDNN is properly installed and running on your Linux system, you will see a message similar to the following:
```
Test passed!
```

参考资料：https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#download

cuDNN下载地址：https://developer.nvidia.com/cudnn