在Ubuntu 18上安装cuDNN

最新推荐文章于 2025-10-14 19:22:23 发布

原创最新推荐文章于 2025-10-14 19:22:23 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

数据挖掘专栏收录该内容

5 篇文章

订阅专栏

本文详细介绍了在Linux系统上安装NVIDIA cuDNN的步骤，包括解压cuDNN包、复制文件到CUDA Toolkit目录并更改权限，以及通过编译和运行mnistCUDNN样本验证cuDNN是否正确安装和运行。

部署运行你感兴趣的模型镜像

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

您可能感兴趣的与本文相关的镜像