如何在 Google Colab 上将 cuDNN 8.1 版降级到 7.6 版

Question

我目前正在做一个仅使用 PaddleOCR 并且仅在 Google Colab 上的 GPU 中运行的 OCR 技术项目。最近，Google Colab 将其 cuDNN 版本从 7.6 更新到了 8.1，导致我的训练模型无法工作。错误消息如下所示。请问有没有办法降级cuDNN版本，或者有没有办法解决这个错误问题？

---

C++ Traceback（最近一次调用最后）：

桨中没有堆栈跟踪，可能是由外部原因引起的。

---

错误信息摘要：

FatalError: Segmentation fault 被操作系统检测到。 [TimeInfo: *** 在 1665370329（unix 时间）中止，如果您使用 GNU 日期，请尝试“date -d @1665370329”***] [SignalInfo: *** PID 523 (TID 0x7f4afdf67780) 从 PID 0 接收到的 SIGSEGV (@0x0) ***]

Answer 1

我遇到了同样的问题（我认为），并且能够通过重新安装此处桨网站上指定的正确软件包来解决。

1. 通过以下任一方式检查 google colab 的 CUDA 版本：

   !nvcc --版本 或者 !nvidia-smi

   2. 转到https://www.paddlepaddle.org.cn/en 并选择适当的信息，包括 CUDA 版本。在 17OCT2022 对我来说，它是 11.2 版。

   3. 修改提供的安装代码以在 google colab 上工作，如下所示：

   !pip install paddlepaddle-gpu==2.3.1.post112 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html

   4. 进入单元格确认 paddlepaddle-gpu 安装正确： 进口桨 paddle.utils.run_check()

   （第 4 步不会告诉您是否根据 CUDA 安装了正确的版本，但它至少会让您确信软件包在其他方面是正确的）