以前有很多关于此主题的问题,但是没有一个答案适合我的具体问题。
我正在尝试在Google Colab上使用Keras 2.2.5训练Mask RCNN。我收到一条消息,提示“您已连接到GPU运行时,但未使用GPU”。我在堆栈溢出时查找了类似的问题:
Why isn't my colab notebook using the GPU?
Anyone experienced the warning about Google colaboratory:You are connected to a GPU runtime, but not utilizing the GPU
,但是所有人都只说要使用使用GPU的软件包,例如Tensorflow。但是,我正在使用与GPU兼容的Keras 2.2.5(大概是Tensorflow 1.14后端,因为我必须安装Tensorflow 1.14才能使Keras 2.2.5正常工作)。发生这种情况有什么原因吗?
更多信息:
- Google Colab
- Python 3.6
- 我使用的是Keras 2.2.5(搭配Tensorflow 1.14)而不是Keras 2.4.3,因为我使用的是2017年的一些代码,仅适用于Tensorflow 1.因此,无法使用Tensorflow 2。
- 我以前曾经使用Google Colab训练GAN,但是从未出现此错误。但是,在这种情况下,我使用了Tensorflow 2。
- 我从来不需要添加任何代码来指定应该运行GPU,并且我不知道是否应该在这里这样做。
- 我必须下载Keras 2.2.5和Tensorflow 1.14才能替换现有版本(Keras 2.4.3和Tensorflow 2.2),我不清楚是否与它有任何关系