一、查看系统现有 cuDNN
1. 登录 高性能应用服务 HAI 控制台。
2. 选择需要连接的算力,然后单击算力连接。在下拉菜单中,单击 JupyterLab。
3. 进入 JupyterLab 后,选择 Other > Terminal,进入终端界面。
4. 在终端输入如下命令,查看系统实际安装的 cuDNN 的版本。
dpkg -l | grep cudnn
得到的终端输出如下,我们可以读出,这个例子中,系统当前安装的 cuDNN 的版本为8.5.0。
二、查找可安装的 cuDNN
在终端输入如下命令,查找可安装的 cuDNN 版本。
apt-cache policy libcudnn8
得到以下输出,在其中找到和系统 CUDA 版本适配的 cuDNN 版本。假设系统 CUDA 已经更新至 11.8 版本,我们可以选择版本 8.9.7 为的 cuDNN。
三、安装指定版本的 cuDNN
1. 根据上一步查找得到的 cuDNN 版本,在命令行中输入如下命令。
apt-get updateapt-get install libcudnn8=8.9.7.29-1+cuda11.8
2. 安装中途,如果遇到询问,输入 y 即可。
3. 完成安装后,再次在终端输入dpkg -l | grep cudnn,返回如下结果,可以看到系统 cuDNN 的版本已经更新为8.9.7。