AssertionError: CUDA unavailable, invalid device 0 requested

yolo를 쓰다가 발생한 오류이다.

 

해결방법

torch 버전을 cuda에 맞게 변경해준다.

pip3 install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

 

맞는 버전은 아래 링크에서 찾아서 쓰면 된다.

 

https://pytorch.org/get-started/previous-versions/

Previous PyTorch Versions