AssertionError: Torch not compiled with CUDA enabled

대다수 Pytorch, Cuda 간 버전 차로 인한 문제

nvcc -V, nvidia-smi 등으로 CUDA 버전을 확인 한 후, 알맞은 Pytorch를 설치한다.

 

아래처럼 간단하게 python 테스트했을 때, True가 리턴되어야 함

>>> import torch
>>> print(torch.cuda.is_available())
True

설치 방법은 아래 command 참조(Version은 다음 링크 참고) ==> https://download.pytorch.org/whl/torch_stable.html

https://download.pytorch.org/whl/torch_stable.html

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html