RuntimeError: The detected CUDA version (12.3) mismatches the version that was used to compile PyTorch (11.8). Please make sure to use the same CUDA versions.
오픈 소스 코드를 받아, 가상 환경 구성 중에 위와 같은 에러가 발생했음
사용 중인 gpu 는 RTX 3090 으로 nvidia-smi 를 찍어보면 cuda 12.2 로 나오고, pytorch 는 cu11.8 로 받아둔 상태
그럼 도대체 12.3 cuda version 은 어디서 오는가??
좀 알아보니 가상환경에 cuda toolkit 이 깔려있지 않아, 기본적으로 깔려있는 cuda version 으로 잡힌 것 같음
그래서 아래 링크에서 11.8 버전을 설치하니 해결
'Code > Pytorch' 카테고리의 다른 글
[Pytorch, timm] Optimizer & Parameter Group Learning Rate (0) | 2023.06.27 |
---|---|
[Pytorch] Batch 개수, Data 개수 (0) | 2023.05.14 |
[Pytorch] Seed 고정 (0) | 2023.05.09 |
[Pytorch] GPU에 맞는 CUDA version 설치 (0) | 2023.04.14 |
[Pytorch] pytorch version 확인 (0) | 2023.04.14 |