Traceback (most recent call last): File "E:\yolov5-master\train.py", line 642, in <module> main(opt) File "E:\yolov5-master\train.py", line 517, in main device = select_device(opt.device, batch_size=opt.batch_size) File "E:\yolov5-master\utils\torch_utils.py", line 118, in select_device assert torch.cuda.is_available() and torch.cuda.device_count() >= len(device.replace(',', '')), \ AssertionError: Invalid CUDA '--device 3' requested, use '--device cpu' or pass valid CUDA device(s)
时间: 2024-04-01 21:35:01 浏览: 95
ant-learn-python-concurrent-main.zip
这是一个错误信息,出现在使用 YOLOv5 进行训练时。错误的原因是选择了一个无效的 CUDA 设备,可能是因为设备号错误或者没有可用的 GPU。建议检查一下 CUDA 设备的设置,确保设备号正确并且该设备可用。如果您没有可用的 GPU,可以选择使用 CPU 进行训练,通过设置 '--device cpu' 来实现。
阅读全文