SSD模型训练过程中CPU占用过高_为什么在gpu上训练模型,cpu也有占用

作者：IT小白 | 2024-05-22 22:42:12

踩

为什么在gpu上训练模型,cpu也有占用

SSD模型训练过程中CPU占用过高

最近在服务器上训练SSD模型，发现模型训练很慢，50000个iter需要16+h。而且CPU占用巨高，把服务器卡爆了被学长说了。但是发现程序确实是挂在GPU上的，GPU占用也正常。上网找了一堆博客，发现问题出在SSD的数据增广模块。他用opencv做的图像裁剪等都是用的CPU，所以很慢，占用也很高。
在这里插入图片描述
参考这篇博客https://blog.csdn.net/qq_27095227/article/details/120728547
因此限制opencv的cpu核心就可以解决问题。在augmentation.py中加入下面设置：

import cv2
cv2.setNumThreads(1)	# 限制opencv占用的线程数
1
2

然后发现效率大大提高，而且cpu占用很低。
在这里插入图片描述
又可以快乐的调参了。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/IT小白/article/detail/610220