在GPU上使用google colab,但现在无法正常工作

时间:2019-11-20 17:35:28

标签: python gpu google-colaboratory hardware-acceleration

我使用Google Colab已有数月了,并提供GPU支持。直到几天前,它说我有GPU支持。现在,事实并非如此。我正在使用相同的代码,并且已经正确设置了硬件加速器。我想知道从Google的角度看我做错了什么。我是一名大学教授,试图为可能的课程学习深度学习知识。因此,我不确定该怎么做。

1 个答案:

答案 0 :(得分:1)

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

%tensorflow_version 1.x

对于tensorflow 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和驱动程序不兼容。