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

문제 상황

pandas를 pip로 설치한 이후에 다시 train 코드를 돌리려니 다음과 같은 에러가 발생했다.

raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

그래서 내 오랜만에 내 가상환경에 어떤 것들이 설치되어 있었는지를 재 점검해보았다.

conda list

............................................................

pytorch                   1.4.0           cpu_py36ha775e86_0
torchvision               0.5.0           pypi_0    pypi

............................................................

pytorch와 torchvision이 멀쩡히 잘 설치되어 있었다.

 

파이썬 버전도 확인해보았다.

python -V

Python 3.6.10 :: Anaconda, Inc.

보니까 파이썬도 3.6으로 잘 설치되어있다.

 

그리고 cuda 또한 잘 설치되어있다.

GPU가 없는 컴도 아닌데 이런 에러가 난다는 것에 매우 놀라면서(논문 써야해서, 우분투에서 윈도우로 갈아타서 쓰는 중인데,

새삼스럽게 윈도우는 이런 일도 있구나 싶다)

 

해결을 위한 시도

에러 해결을 위해 서치를 해보니, 이 문제는 아무래도 pytorch의 버전에서 생기는 에러였던 듯 하다.

그래서 서치한 사이트에서 발견한 나와 같은 버전을 가진 사람 - 윈도우 10, CUDA v10.0를 가진 사람의 코멘트를 따라 다음과 같은 명령어를 입력해주었다.

pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html

이렇게 되면 기존의 torch버전과 torchvision의 버전이 0.1 씩 올라가게 되는 것이다.

 

 

 

참고한 사이트

github.com/pytorch/pytorch/issues/30664