AssertionError ("Torch not compiled with CUDA enabled")

상황

아나콘다 가상환경에서 오픈 소스를 clone해서 구동하는 과정에서 다음과 같은 에러가 발생했다.

File "/home/ksw/anaconda3/envs/coca-pytorch/lib/python3.11/site-packages/torch/cuda/__init__.py", line 289, in _lazy_init
    raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

찾아보니 다음 글에서 원인을 찾을 수 있었다.

• non-GPU enabled version of Pytorch installed, as the error is clearly telling you. You need to install a GPU enabled version.

pip install torch로만 torch를 예전에 설치했었는데 그것이 문제가 되었던 것이다.

해결

파이토치 공식 홈페이지에서 cuda version에 맞는 conda 명령어를 통해서 해결했다.

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