728x90
RuntimeError: The detected CUDA version (10.1) mismatches the version that was used to compile PyTorch (11.3). Please make sure to use the same CUDA versions
-
RuntimeError: The detected CUDA version (10.1) mismatches the version that was used to compile PyTorch (11.3). Please make sure to use the same CUDA versions검색하기 귀찮아서 블로그에 박제 2023. 7. 13. 14:06
RuntimeError(CUDA_MISMATCH_MESSAGE.format(cuda_str_version, torch.version.cuda)) 가상환경에서 nvcc -V 했을 때 버전이 안바껴서 그럼 (base)환경에서의 버전이 보임 그래서 nvcc 버전을 바꿔줘야함 $vi ~/.bashrc 마지막에 추가 export CUDA_HOME="/usr/local/cuda-11.2" export LD_LIBRARY_PATH="/usr/local/cuda-11.2/lib64:$LD_LIBRARY_PATH" export PATH="/usr/local/cuda-11.2/bin:$PATH" $source ~/.bashrc