윈도우 파이토치 쿠다 버전 안 맞는 오류 해결법 AssertionError : torch not compiled with cuda enabled

최근 Windows 환경에서 Pytorch를 GPU버전으로 설치하려는데, torch는 실행이 잘 되는데 GPU와의 연결이 잘 안되는 경우가 생겼다.

에러 메시지는 다음과 같다. (리눅스에서도 뜨는 경우를 봤던 것 같다)

AssertionError : torch not compiled with cuda enabled

 

참고로 pytorch 설치는 다음과 같이 진행하였다.

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

 

이 중 한 깃헙 이슈에서 발견한 좋은 해결책을 공유한다.

pip uninstall torch
pip cache purge
pip install torch -f https://download.pytorch.org/whl/torch_stable.html

사실 torch를 제거하고 다시 설치하는 것으로 간단하긴 한데, 간략하게 설명을 하자면

 

우선 uninstall을 진행하고, 두 번째 줄은 pip에 있는 cache를 없애는 것이다. 이게 uninstall을 하고 다시 설치하더라도 pip에서 기존에 있었던 또 다른 이상한 버전의 패키지를 설치할 수도 있어서 이를 방지하는 과정이라고 보면 된다. 또 세번째 줄은 stable한 버전을 다시 설치하는 것이다.

 

이 세 줄만 입력하니 문제 없이 잘 돌아갔고, 인터넷을 보다 보니 오류를 고치기 위해서 재부팅, 껐다 키는 것은 의미 없다는 글을 보았다. 저 세 줄로 문제를 해결하는 과정에서도 재부팅은 하지 않았다.