在 Vertex AI Notebook 中升级 Cudnn 版本答案

【问题标题】：Upgrading Cudnn version in Vertex AI Notebook在 Vertex AI Notebook 中升级 Cudnn 版本
【发布时间】：2023-01-20 21:48:02
【问题描述】：

问题:Cudnn版本不兼容tensorflow和Cuda, 内核死机并且无法在 Vertex AI 中开始训练。

当前版本：

import tensorflow as tf
from tensorflow.python.platform import build_info as build
print(f"tensorflow version: {tf.__version__}")
print(f"Cuda Version: {build.build_info['cuda_version']}")
print(f"Cudnn version: {build.build_info['cudnn_version']}")

tensorflow version: 2.10.0
Cuda Version: 11.2
Cudnn version: 8

根据 here 提供的信息（如所附屏幕截图所示），Cudnn版本必须是 8.1。

here有人问过类似的问题，跟升级有关Cudnn在谷歌协作中。但是，它并没有解决我的问题。所有其他在线资源仅对 Anaconda 环境有帮助。

我怎样才能升级Cudnn在我的情况下？

谢谢你。

【问题讨论】：

Google 的article 是否可以帮助您解决问题？让我知道这是否有帮助。

标签： tensorflow cuda google-cloud-vertex-ai cudnn

【解决方案1】：

我在 Google Colab 中尝试了几种 tensorflow、Cuda 和 Cudnn 版本的组合，以下版本有效 [操作系统：Ubuntu 20.04]：

tensorflow version: 2.9.2
Cuda Version: 11.2
Cudnn version: 8

因此，我将 Vertex AI 中的 tensorflow 版本从 2.10.0 降级到 2.9.2 并且它起作用了（仅解决了不兼容问题）。我仍在寻找内核重启的解决方案。

更新：：

更改内核后，内核重新启动的问题得到解决Tensorflow 2 (Local)到Python (Local)在 Vertex AI 的笔记本中，如附图所示 [内核更改选项在错误符号附近的右上角可用]。

【讨论】：