쿠다 에러 CUDA error: no kernel image is available for execution on the deviceCUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.

Stability Matrix에서 AI 툴킷에서 로라 학습을 하려고 했는데 위와 같은 에러가 떴습니다. 

 

이번에 컴퓨터를 바꾸면서 5080으로 구입했는데 AI이미지를 만들어보고자 comfyui도 설치하고 로라를 직접 만들어보려고 했습니다. 

 

그런데 제목과 같은 에러가 뜨면서 학습이 안되더라구요. 그래서 한참 여기저기 구글링해서 해결한 결과를 공유하고자 합니다.

 

1. 설치 전 환경 정리 (필수)

 

win+r 버튼으로 cmd를 관리자 권한으로 엽니다. 그런 후에 이전에 설치했던 PyTorch, 특히 문제가 되었던 이전 버전이나 특별 빌드가 남아있으면 충돌이 날 수 있습니다. 아래 명령어 복사 붙여넣기 하시면 됩니다. 

# 기존 PyTorch 및 관련 라이브러리 제거
pip uninstall torch torchvision torchaudio

2. 환경 변수 설정 (안전 조치)

PyTorch 2.7.0이 CC 12.0 (Blackwell)을 공식적으로 지원하더라도, 혹시 모를 상황에 대비해 GPU Compute Capability를 명시적으로 설정하고 학습을 시작하는 것이 좋습니다.

• Windows (Command Prompt):RTX 5080은 CC 12.0이지만, PyTorch 환경 변수에서는 9.0이 최신 아키텍처를 포괄하는 표준 값으로 사용되는 경우가 많습니다.

• set TORCH_CUDA_ARCH_LIST="9.0"
  

3. 설치 실행

위에서 확인한 PyTorch 2.7.0 설치 명령어를 실행합니다.

pip install torch==2.7.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

 

 

마무리. 

위의 방법대로 했더니 에러 없이 지금 로라를 학습중입니다.