lbt162020 2022-03-08 10:24 采纳率: 68.8%
浏览 251
已结题

pytorch训练时怎么提高显卡的利用率？

img


如图，我的2060才利用了百分之十几，跑个简单的模型都慢死。想问一下怎么提高利用率？

  • 写回答

2条回答 默认 最新

  • 爱晚乏客游 2022-03-08 10:29
    关注

    work number设置到cpu最大核心数，batch size 搞大点，你的显存才用了2G，还有那么多呢，不过显卡利用率低很正常，又不是挖矿，显卡很多时候要等硬盘或者内存读取信息呢。
    你把这里换成cuda就知道了，一阵高一阵低，就是显卡计算快，数据传输慢，导致显卡计算完了要等其他数据才能继续计算

    img

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(1条)

报告相同问题？

问题事件

  • 系统已结题 3月16日
  • 已采纳回答 3月8日
  • 创建了问题 3月8日