2024-04-25 06:52:13 发布
网友
我使用的Mask RCNN库是基于tenserflow的,我似乎无法让它在我的GPU(1080TI)上运行。推断时间是4-5秒,在此期间,我看到cpu上出现了使用高峰,但gpu上没有。有什么解决办法吗?在
这可能是因为config.py中的GPU计数设置为0,或者您没有安装tensorflow-gpu(这是tensorflow在GPU上运行所必需的)
config.py
tensorflow-gpu
这可能是因为
config.py
中的GPU计数设置为0,或者您没有安装tensorflow-gpu
(这是tensorflow在GPU上运行所必需的)相关问题 更多 >
编程相关推荐