Mask-RCNN使用CPU而不是GPU

2024-04-25 06:52:13 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用的Mask RCNN库是基于tenserflow的,我似乎无法让它在我的GPU(1080TI)上运行。推断时间是4-5秒,在此期间,我看到cpu上出现了使用高峰,但gpu上没有。有什么解决办法吗?在


Tags: gpu时间maskcpurcnn解决办法tenserflow