cuda오류

꿀팁/오류 정리

A100-SXM4-40GB with CUDA capability sm_80 is not compatible with the current PyTorch installation.

원인 pytorch 버전이 CUDA sm_80 아키텍처를 지원하지 않기 때문에 발생하는 오류입니다. 해결 방법 pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html CUDA sm_80버전을 지원하는 pytorch를 설치하시면 됩니다.

컴공누나
'cuda오류' 태그의 글 목록