1.按照requirements.txt安装时torch和torchvision要注释掉,因为会默认安装cpu版本,直接官网安装pytorch
2.numpy库最好安装1.24之前的版本,1.24之后np.int被删除,需要改成np.int_
3.utils/loss.py中gain = torch.ones(7, device=targets.device)需要改为gain = torch.ones(7, device=targets.device).long(),因为新版本的torch无法自动执行此转换
4.确定用到GPU训练后,遇到爆显存情况可以调整train.py中的batchsize和numworkers
batchsize以2的倍数往上加,num_workers可以从1开始,和cpu核数没有具体关系但可以作为参考
Yolov5官方代码运行问题
于 2023-06-15 21:14:33 首次发布