谷歌 colab 上的 cuDNN

Question

当我运行以下行时：

import tensorflow

我收到如下错误：

导入错误：libcudnn。所以。 6：无法打开共享对象文件：没有这样的文件或目录

我在我的项目中使用 Google Colab。我正在使用tensorflow-gpu 1.4。不知何故，我设法在 Google Colab 上安装了Cuda 8.0。但是我需要安装cuDNN，这是tensorflow-gpu 1.4的必要条件。我不应该将 tensorflow 版本升级到 1.12。

如何在 Google Colab 上安装 cuDNN？

Answer 1

获取正确使用 GPU 的各种包的版本很复杂，这就是 Colab 为您做这件事的原因。如果您尝试使用另一组版本，您将会遇到麻烦，但如果您真的想尝试，那么答案是按照 NVIDIA 的文档了解如何安装他们的东西。

请注意，由于用户态库和驱动程序版本不是独立的，因此您无法在 colab 上更改驱动程序版本。