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(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