크크루쿠쿠

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

StackOverflow

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

JH_KIM 2022. 2. 24. 10:05

RTX 30시리즈의 경우에는 CUDA 버전이 11.0 이상만 호환이 되기 때문에 나의 경우에는 torch cuda version이 10 버전을 다운받아서 생긴 문제였다.

아마 torch 1.7 버전부터 cuda 11.0 을 지원하는것으로 알고 있다.

그러므로 설치할 때

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html

이렇게 cuda 버전을 명시해줘야 에러를 면할 수 있다.

Comments