我在使用基于 TensorFlow 的 Mask RCNN 库,但似乎无法在我的 GPU(1080TI)上运行。推理时间为 4-5 秒,在此期间我看到 CPU 使用率激增,但 GPU 没有。有什么可能的解决方法吗?
回答:
这要么是因为在 config.py
中 GPU_COUNT 被设置为 0,要么是因为你没有安装 tensorflow-gpu
(这是 TensorFlow 在 GPU 上运行所必需的)。
我在使用基于 TensorFlow 的 Mask RCNN 库,但似乎无法在我的 GPU(1080TI)上运行。推理时间为 4-5 秒,在此期间我看到 CPU 使用率激增,但 GPU 没有。有什么可能的解决方法吗?
回答:
这要么是因为在 config.py
中 GPU_COUNT 被设置为 0,要么是因为你没有安装 tensorflow-gpu
(这是 TensorFlow 在 GPU 上运行所必需的)。