我的GPU型号是NVIDA GeForce RTX 3060 在训练ephoch=300,大概两万张图片,训练了将近20个小时。
发现租用GPU进行模型训练一方面的原因是不用配置环境,节省很多麻烦事。因为用cpu训练非常的慢,然后配置GPU的环境,配置cuda和cudann,加上一些版本,驱动,兼容等问题,需要花很长时间。另一方面,长期用自己的笔记本电脑训练模型的话,电脑的寿命减少的很快。
下面给大家推荐两个我自己用的很好用的租用GPU的平台。理由在于数据集的上传速度够快,11个G的数据集不会超过30min,2个G的数据不会超过1min。价格也足够实惠。
1.Gpushare Cloud
(1)新人注册,完成任务有50元的代金券。
2.FeaTurize
其中Featurize,我在训练YOLOV7.0的时候,还是很顺利的,稍微修改下.yaml文件还有train.py轻松开训练。但是在使用恒源云的时候,训练有些麻烦,数据集不能够放在hy-tmp文件中(数据储存),放在那里面会一直报错读取不到数据集。其次,路径问题,路径不能设置绝对路径,还有竟然会出现不能下载YOLO字体而报错,只能自己先下载好,然后在放进去。
然后FeaTurize我写了一个教程,给大家参考,方便大家快速上手。