AssertionError: Invalid CUDA '--device 1' requested, use '--device cpu' or pass valid CUDA device(s)

[10756007 (維辰 林)]

YOLOv5 2022-3-25 torch 1.8.1+cu111 CUDA:0 (NVIDIA GeForce RTX 3060 Laptop GPU, 6144MiB) -> Invalid CUDA '--device 0'

I’m trying to train yolov5s with bdd100k on my GPU locally and getting this error although my GPU check is correctly done. I installed latest version of cuda and pytorch. ERROR: github: up to date with GitHub - ultralytics/yolov5: YOLOv5 in PyTorch > ONNX > CoreML > TFLite fatal: cannot change to ‘C:\Users\Luka\Desktop\Berkeley’: No such file or directory Traceback (most recent call last): File “C:\Users\Luka\Desktop\Berkeley dataset\yolov5s_bdd100k\yolov5\train.py”, line 643, in main(op…

I have the same problem as him