我在运行YOLOV5 Lite的train.py的时候。出现这个错误应该咋办?
github: skipping check (not a git repository)
Traceback (most recent call last):
File "D:\YOLOv5-Lite\train.py", line 523, in <module>
device = select_device(opt.device, batch_size=opt.batch_size)
File "D:\YOLOv5-Lite\utils\torch_utils.py", line 71, in select_device
assert torch.cuda.is_available(), f'CUDA unavailable, invalid device {device} requested' # check availability
AssertionError: CUDA unavailable, invalid device 0/cpu requested
ERROR conda.cli.main_run:execute(124): `conda run python D:\YOLOv5-Lite\train.py` failed. (See above for error)
进程已结束,退出代码为 1