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

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

报告相同问题？

关注问题

pytorch训练时怎么提高显卡的利用率？ pytorch 深度学习自然语言处理
2022-03-08 10:24

回答 2 已采纳 work number设置到cpu最大核心数，batch size 搞大点，你的显存才用了2G，还有那么多呢，不过显卡利用率低很正常，又不是挖矿，显卡很多时候要等硬盘或者内存读取信息呢。你把这里换成c
Windows版本，AMD显卡能搞Pytorch吗？ pytorch 深度学习神经网络
2022-06-08 22:43

回答 1 已采纳 ROCm路子官网只支持linux，不过我也没用过，但是听说bug和问题很多，另外就是微软的pytorch-DirectML，你可以试试看下这个能不能走得通，走不通就没办法了,深度学习还得是cuda。
pytorch训练时cuda内存不足 pytorch 目标检测神经网络
2023-03-02 15:28

回答 2 已采纳通常遇到OOM(out of memory)问题, 只有两种解决方案, 降低您网络训练时的batchsize, 或者选用更小的网络. 看到您这里用的resnet50, 8Gmemory够用了, 您可以
PyTorch GPU 利用率低？常见原因分析及优化
2022-07-16 14:57

机器学习与AI生成创作的博客转载自腾讯技术工程一、GPU 利用率的定义本文的 GPU 利用率主要指 GPU 在时间片上的利用率，即通过 nvidia-smi 显示的 GPU-util 这个指标。统计方式为：在采样周期内，GPU 上面有 kernel 执行的时间百分比。二...
没有解决我的问题, 去提问