当前位置:   article > 正文

(14)关于PyTorch训练深度学习模型CPU/GPU使用率低,训练速度慢这一问题的总结_训练中cpu占用率低怎么解决

训练中cpu占用率低怎么解决

电脑配置双2080Ti+i7-10700K,使用mobilenet-yolov4-lite训练VOC数据集时,训练100epochs要五六个小时,检测cpu和gpu的使用发现利用率都很低,说明没有充分使用电脑优秀的性能,两个显卡的使用率都分别只有20%-50%,查阅一些资料总结如下:

1、参考:【深度学习】踩坑日记:模型训练速度过慢,GPU利用率低

上面作者指出四个提速技巧:

  • 减少日志IO操作频率
  • 使用pin_memory和num_workers
  • 使用半精度训练
  • 更好的显卡,更轻的模型

其中,第四个无力改变,前两个均进行尝试,首先在train.py文件中改变了日志输出

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/102745
推荐阅读
相关标签
  

闽ICP备14008679号