Google Colab GPU：[]

Question

我正在 google colab 中使用 maskrcnn 模型。我从运行时部分选择 GPU。

import tensorflow as tf
print(tf.__version__)
print(tf.test.gpu_device_name())
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

输出：

2.18.0-dev20240613
/device:GPU:0
Num GPUs Available:  1

我检查过这里没有问题。

当我设置maskrcnn的要求时， !pip install -r /content/maskrcnn/requirements.txt

2.14.0

Num GPUs Available:  0

GPU 消失了。

需求.txt

cython==3.0.5
h5py==3.9.0
imgaug==0.4.0
ipython==7.34.0
ipython-genutils==0.2.0
ipython-sql==0.5.0
keras==2.14.0
matplotlib==3.7.1
numpy==1.23.5
opencv-contrib-python==4.8.0.76
opencv-python==4.8.0.76
pillow==9.4.0
scikit-image==0.19.3
scipy==1.11.3
tensorboard==2.14.1
tensorflow==2.14.0

我不明白为什么GPU会消失。有人有什么想法吗？ 谢谢您，最美好的祝愿。

Answer 1

查看您的

requements.txt

有

tensorflow==2.14.0

- 因此它可以卸载您的

tensorflow 2.18

（适用于 GPU）并安装

tensorflow 2.14.0

，这可能与 GPU 有问题。

所以你必须再次安装

tensorflow 2.18

。