728x90
AssertionError: CUDA unavailable invalid device 0 requested
-
AssertionError: CUDA unavailable, invalid device 0 requested검색하기 귀찮아서 블로그에 박제 2021. 7. 19. 13:24
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 Installing previous versions of PyTorch pytorch.org