AssertionError: Torch not compiled with CUDA enabled

 

현재 깔려 있는 cuda version 은 11.3, 프로젝트에 써야할 버전도 11.3 

 

pytorch 공식 문서에 있는 

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

해도 계속 되는 cuda enabled assertion error.

 

1. 

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

이걸로 하니까 해결... 

 

2. 어디선가 패키지끼리 버전이 꼬여서 생기는 문제이기도 해서,

가상환경 자체를 삭제하고 다시 처음부터 package들 install 하면 되기도 함.