标签： python tensorflow machine-learning keras
我正在使用基于tenserflow的Mask RCNN库，我似乎无法使其在我的GPU（1080TI）上运行。推理时间为4-5秒，在此期间，我看到我的cpu而不是我的gpu的使用高峰。有任何可能的解决方法吗？
答案 0 :(得分：2)
要么是因为config.py中的GPU_COUNT设置为0，要么您没有安装tensorflow-gpu（这是Tensorflow在GPU上运行所必需的）
config.py
tensorflow-gpu