gpt4 book ai didi

python - Mask RCNN 使用 CPU 而不是 GPU

转载 作者:行者123 更新时间:2023-11-30 08:57:16 25 4
gpt4 key购买 nike

我正在使用基于tenserflow的Mask RCNN库,但我似乎无法让它在我的GPU(1080TI)上运行。推理时间为 4-5 秒,在此期间我看到 CPU 使用量激增,但 GPU 却没有。有什么可能的解决办法吗?

最佳答案

这要么是因为 config.py 中的 GPU_COUNT 设置为 0,要么是您没有安装 tensorflow-gpu(这是 TensorFlow 运行所必需的) GPU)

关于python - Mask RCNN 使用 CPU 而不是 GPU,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55211277/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com