在 google colab 中将 CUDNN 升级到 8.2

Posted 2023-02-16

【中文标题】在 google colab 中将 CUDNN 升级到 8.2

【英文标题】：Upgrade CUDNN to 8.2 in google colab

【发布时间】：2021-12-29 02:20:12

【问题描述】：

我想使用将 CUDNN 版本从 8.0 升级到 8.1 并将 CUDA 版本升级到 11.2，但我不确定我们如何在 colab 上做到这一点。下面是我编写的在 google colab 上运行的脚本，但它没有任何改变。我需要在这里进行哪些更改？或者有其他方法吗？

!sudo apt install cuda-toolkit-11-2
!wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin
!sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600
!sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
!sudo add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/ /"
!sudo apt-get update
!sudo apt-get -y install cuda
!sudo apt --fix-broken install
!sudo apt install cuda-cudart-11-2
import os
os.environ['PATH'] = '/usr/local/cuda-11.2/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/tools/node/bin:/tools/google-cloud-sdk/bin:/opt/bin'
os.environ['LD_LIBRARY_PATH'] = '/usr/local/cuda-11.2/lib64:/usr/local/cuda-11.2/lib64:/usr/lib64-nvidia'
!cp "/content/drive/My Drive/cudnn-11.2-linux-x64-v8.1.0.0.tgz" "/content/cudnn-11.2.tgz"
!mkdir cudnn
!tar -C cudnn/ -zxvf cudnn-11.2.tgz
%cd cudnn/
!sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.2/include
!sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.2/lib64
!sudo chmod a+r /usr/local/cuda-11.2/include/cudnn*.h /usr/local/cuda-11.2/lib64/libcudnn*
!sudo ldconfig
%cd /content/

检查 CUDA 和 CUDNN 版本

! nvcc --version
!nvidia-smi

import chainer

print("-----------------SYSTEM INFO----------------------")
chainer.print_runtime_info()

我正在尝试使用 Tensorflow 对象检测并按照此链接检查 cuda 和 cudann 的兼容版本：link

【问题讨论】：

尝试做同样的事情。收到此错误消息：已加载运行时 CuDNN 库：8.0.5 但源代码编译为：8.1.0。 CuDNN 库需要具有匹配的主要版本和相等或更高的次要版本。如果使用二进制安装，请升级您的 CuDNN 库。

【参考方案1】：

只需将运行时类型更改为 TPU。这对我有用

【讨论】：

这可能适用于一小组数据。但最终你需要使用 Nvidia GPU 芯片来处理大量数据。