运行yolo5出现的问题大家可以帮吗看一下吗

Question

yolo5训练模型出现的报错

Traceback (most recent call last):
  File "C:/Users/13789/Desktop/trffic_light/window.py", line 525, in <module>
    mainWindow = MainWindow()
  File "C:/Users/13789/Desktop/trffic_light/window.py", line 51, in __init__
    self.model = self.model_load(weights="runs/train/exp23/weights/best.pt",
  File "C:\Users\13789\.conda\envs\pytrffic\lib\site-packages\torch\autograd\grad_mode.py", line 27, in decorate_context
    return func(*args, **kwargs)
  File "C:/Users/13789/Desktop/trffic_light/window.py", line 65, in model_load
    device = select_device(device)
  File "C:\Users\13789\Desktop\trffic_light\utils\torch_utils.py", line 65, in select_device
    assert torch.cuda.is_available(), f'CUDA unavailable, invalid device {device} requested'  # check availability
AssertionError: CUDA unavailable, invalid device 0 requested

Answer 1

你是不是没有安装cuda 或者你电脑没有gpu