python - 使用带有 GPU 的 google colab，但现在无法正常工作

Question

我已经使用 Google Colab 几个月了，现在支持 GPU。直到几天前，它说我有 GPU 支持。现在，它没有。我正在使用相同的代码，并且我已经正确设置了硬件加速器。我想知道从谷歌的角度来看我是否做错了什么。我是一名大学教授，试图为可能的课程学习深度学习内容。所以，我不确定我能做些什么来实现这一点。

标签： pythongpugoogle-colaboratoryhardware-acceleration

Answer 1

在 Colab 上使用 tensorflow 时，建议使用捆绑的预安装版本。Tensorflow 1.X 和 2.X 可用，并且可以通过魔术函数启用：

%tensorflow_version 1.x

对于张量流 1，和

%tensorflow_version 2.x

对于 tensorflow 2（有关详细信息，请参阅https://colab.research.google.com/notebooks/tensorflow_version.ipynb）。

如果您通过pip install tensorflow-gpu==2.0或类似方式安装 tensorflow 的外部版本，它将安装一个预构建的二进制文件，该二进制文件可能与 Colab 运行时中可用的 GPU 和驱动程序不兼容。