pytorch吧 关注:1,456贴子:2,578
  • 1回复贴,共1

pytorch跑程序gpu利用率变成0

只看楼主收藏回复

pytorch跑程序的时候一开始GPU利用率为3%,程序跑得很快。
过了大概半个小时,程序运行得越来越慢,我查了一下,GPU利用率变成0了。
导致我现在处理一个一万多帧的视频用了十几个小时才处理了两千多帧。
我的cuda版本是11.0,pytorch版本也是对应的,不知道哪里出了问题,请问可以帮我看看是哪里出了问题吗?
#Python##pytorch##GPU#





IP属地:广东来自Android客户端1楼2022-03-29 22:58回复
    尝试使用更多的numworker?然后在dataloader里写入pin_memory=true,将数据锁入显存。


    IP属地:福建来自Android客户端2楼2022-04-28 20:36
    回复