K41*_*F4r 2 python machine-learning keras tensorflow
我正在使用基于 tenserflow 的 Mask RCNN 库，但我似乎无法让它在我的 GPU (1080TI) 上运行。推理时间为 4-5 秒，在此期间我看到我的 cpu 出现使用高峰，但我的 gpu 没有。对此有任何可能的修复吗？
要么是因为 GPU_COUNT 设置为 0config.py要么你没有tensorflow-gpu安装（这是 tensorflow 在 GPU 上运行所必需的）
| 归档时间： |
|
| 查看次数： |
3249 次 |
| 最近记录： |