-
현상: CUDA error: no kernel image is available for execution on the device 에러 발생
-
원인: 설치한 CUDA 버전(11.4)와 현재 pytorch와 연결하여 설치된 버전(10.4)이 맞지 않아 생기는 오류. torch.cuda.isavailable()에서도 True를 반환하나 실제 사용이 불가하다.
-
확인 방법: 현재의 pytorch의 cuda 버전을 확인해 본다.
import torch
print(torch.__version__)-
확인 결과: 설치된 cuda는 nvcc -V 로 확인해보면 11.4인데 pytorch의 호환 CUDA 버전은 10.4인 것을 알 수 있다.
-
해결 방법: CUDA 호환 가능한 pytorch를 다시 설치한다.
conda install pytorch torchvision torchaucio cudatoolkit=11.3 -c pytorch
