[error] RuntimeError: CUDA error: no kernel image is available for execution on the device 에러 [Ubuntu 20.04]

[error] RuntimeError: CUDA error: no kernel image is available for execution on the device 

위의 에러는 파이토치 버전과 CUDA 버전의 호환성 에러이다. 

실행 환경 : Ubuntu 20.04 / RTX 3060 / CUDA 10.1

---

1.  먼저 나의 GPU 사양을 확인한다. 

nvidia-smi --query | fgrep 'Product Name'

 

2.  https://developer.nvidia.com/cuda-gpus#compute 로 이동

 

3.  CUDA - Enabled GeForce and TITAN Products 선택

 4.  나의 GPU 에 맞는 Compute Capability 를 확인한다.

 

5.  터미널을 열고 다음 명령어 실행

export TORCH_CUDA_ARCH_LIST=8.6

 

6.  이후 기존에 설치되어 있던 torch 와 torchvision 등을 삭제한다.

pip uninstall torch torchvision

 

7.  나의 CUDA 버전을 확인한다.

# Linux
nvcc -V

# Window
nvcc.__version__

맨 밑의 release 를 보면 된다. 10.1 버전 임을 알 수 있다.

 

 

8.  https://pytorch.org/get-started/previous-versions/ 로 이동

 

9.  나의 CUDA 버전과 맞는 torch 버전을 설치한다. (e.g. CUDA 10.1 버전)

# CUDA 10.1
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html