首页 > 解决方案 > 如何在 Google Colab 中将 cuDNN 从 v8.0 降级到 v6.0?

问题描述

这是从这个问题开始的:如何在 Google Colab 中降级 Cuda 和 cuDNN 版本?

我已按照说明进行操作,并首先将 Cuda 10.0 降级为 Cuda 8.0。然后我将 cuDNN v6.0 下载到我的本地驱动器并上传到我已安装的 Google 驱动器。然后我使用以下代码安装:

!dpkg -i "~/Downloads/libcudnn6_6.0.21-1+cuda8.0_amd64.deb"

!ls -l /usr/lib/x86_64-linux-gnu/libcudnn.so.6*

我在日志中收到如下消息:

/sbin/ldconfig.real: /usr/local/lib/python3.6/dist-packages/ideep4py/lib/libmkldnn.so.0 is not a symbolic link

它似乎已安装:

lrwxrwxrwx 1 root root        18 Apr 12  2017 /usr/lib/x86_64-linux-gnu/libcudnn.so.6 -> libcudnn.so.6.0.21

-rw-r--r-- 1 根 154322864 2017 年 4 月 12 日 /usr/lib/x86_64-linux-gnu/libcudnn.so.6.0.21

当我检查 cuDNN 的安装位置时,我得到:

cudnn: /usr/include/cudnn.h

检查我得到的版本:

#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

但是,当我尝试使用我的模型时,我收到:

no CUDA-capable device is detected

我很困惑。我是否正确安装了正确版本的 cuDNN?

标签: google-colaboratory

解决方案


推荐阅读