1.可用环境
torch==1.7.1+cu110
torchvision==0.8.2+cu110
2.错误环境遇到的问题
(1)情况一
torch==1.7.0+cu110
torchvision==0.8.0
训练yolov5时,训练就会开始报错:
return torch.ops.torchvision.nms(boxes, scores, iou_threshold)
RuntimeError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend. 'torchvision::nms' is only available for these backends: [CPU, BackendSelect, Named, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, Tracer, Autocast, Batched, VmapMode].
torchvision版本不对。 所以升级了torchvision的版本。进入到情况二。
(2)情况二
torch==1.7.0+cu110
torchvision==0.8.1
训练第一个epoch时,不出错,但是测试第一个epoch的时候,会报错: