AssertionError: Torch not compiled with CUDA enabled 에러 해결 방법

에러확인 )

파이썬에서 AI 그림을 생성하는 실습을 하는 중에 AssertionError: Torch not compiled with CUDA enabled 에러가 발생하였습니다.

에러 메시지는 다음과 같습니다:

조치방법 )

위 실습을 하기 위해서는 CUDA(Compute Unified Device Architecture) 및 Torch를 설치해야 실행이 가능합니다. 하지만 설치를 했음에도 불구하고 위와 같은 에러가 발생할 수 있습니다.

이 문제의 원인은 여러 가지가 있는데, 가장 대표적인 이유는 설치된 CUDA와 Torch 버전이 일치하지 않는 경우입니다.

그 외에도 NVIDIA 기반의 GPU가 아닌 경우, 특히 Mac을 사용하는 경우에는 다른 환경 설정이 필요할 수 있습니다.

우선 Window환경으로 CUDA를 설치할때 아래 링크에서 조회되는 명령어로 설치한다.

문제 해결 방법

1. CUDA 및 Torch 버전 확인

• 설치된 CUDA 버전과 Torch 버전이 호환되는지 확인합니다.
• 호환되지 않는 경우, 올바른 버전으로 다시 설치해야 합니다.

1. CUDA 및 Torch 재설치

• Windows 환경에서 CUDA를 설치할 때는 다음 링크에서 조회되는 명령어로 설치합니다: PyTorch 설치 가이드
• 예를 들어, 아래 명령어를 사용하여 CUDA 12.1과 호환되는 Torch를 설치합니다.

명령어 > pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
  

다운로드 링크 << CUDA 다운

아래 기준으로 Torch를 다운받을 때 그림과 같이 버전을 확인 (cu121하여) 해당버전을 설치 해주면 된다.

pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121
  
  추가 조치 사항
  NVIDIA 드라이버 업데이트: 최신 버전의 NVIDIA 드라이버를 설치합니다. 이는 CUDA와의 호환성을 보장합니다.
  환경 변수 설정: CUDA Toolkit이 올바르게 설치되었는지 확인하고, PATH 환경 변수에 CUDA의 bin 디렉토리가 포함되어 있는지 확인합니다.
  PyTorch 설치 확인: PyTorch가 올바르게 설치되었는지 확인합니다. 터미널에서 다음 명령어를 입력하여 CUDA가 활성화된 PyTorch를 확인할 수 있습니다:
  위의 방법을 따라 설정을 완료하면 AssertionError: Torch not compiled with CUDA enabled 에러가 해결될 것입니다.
  
  혹시 다른 방법으로도 해결해본 경험이 있다면, 댓글로 공유해 주세요. 읽어주셔서 감사합니다.