Visual studios can not find cuda Error while training on yolo7v

Question

when i try to train on yolo v7 I get this error:

File "train.py", line 595, in <module>
    device = select_device(opt.device, batch_size=opt.batch_size)
  File "C:\Users\96Crori\Desktop\yolov7_custom_training\yolov7\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 requested

I installed cuda version 11.3 but i dont know why visual studios cant find it

Answer 1

I managed to figure out the problem. I had the wrong version of Pytorch installed which was not letting me access cuda. Once I installed the correct version the error went away.