kimkj38.log
로그인
kimkj38.log
로그인
CUDA capability sm_86 is not compatible with the current PyTorch installation
김경준
·
2022년 8월 2일
팔로우
1
error
tips
1
Error
목록 보기
1/7
- nvcc --version - torch.cuda.get_arch_list() - torch.version.cuda
서버(A5000)의 compute capability에 비해 pytorch 버전이 낮아서 나타나는 문제로 보임
(
https://en.wikipedia.org/wiki/CUDA
)
cuda version 확인 시 11.1
pytorch는 1.6으로 설치했다가 에러 확인 후 uninstall 후 conda install로 1.9 버전 재설치
(
https://pytorch.org/get-started/previous-versions/
)
arch list는 여전히 ['sm_37', 'sm_50', 'sm_60', 'sm_70']
torch.version.cuda는 10.2
다시 uninstall 후 cuda11.1에 맞는 wheel로 설치
(pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f
https://download.pytorch.org/whl/torch_stable.html
)
Reference
https://discuss.pytorch.org/t/nvidia-geforce-rtx-3090-with-cuda-capability-sm-86-is-not-compatible-with-the-current-pytorch-installation/141940/10
김경준
팔로우
1개의 댓글
댓글 작성
최창수
2022년 9월 19일
감사합니다 경준님!!!!!!!!! 사랑해요!!
답글 달기
감사합니다 경준님!!!!!!!!! 사랑해요!!