ubuntu22.04 docker安装cudnn
时间: 2023-09-30 10:04:37 浏览: 79
首先,你需要下载对应版本的CUDNN,官方下载地址:https://developer.nvidia.com/rdp/cudnn-download
接着,按照以下步骤进行安装:
1. 将下载的CUDNN压缩包解压到某个目录下,例如:/home/user/cudnn/
```bash
tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz
```
2. 进入解压后的目录,将文件复制到CUDA的安装目录下,例如:/usr/local/cuda/
```bash
sudo cp -P cuda/include/cudnn*.h /usr/local/cuda/include/
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
```
3. 验证安装是否成功
```bash
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
```
如果输出类似下面的信息,则说明安装成功:
```bash
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 2
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
```
注意:以上操作需要在root权限下执行,或者使用sudo命令。
阅读全文