Torch not compiled with CUDA enabled

torch와 cuda 버전이 안맞아서 나는 에러인듯

 

conda install -c pytorch torchvision cudatoolkit=10.1 pytorch

와 같은 명령어로 해결될수 있음(cudatoolkit과 원래 깔려있던 쿠다 버전이 맞을경우 -> 알맞은 torch 버전 깔아줌)

 

https://pytorch.org/get-started/locally/

PyTorch

위 사이트에서 콘다에 깔기 원하는 쿠다 버전과 원하는 torch 버전을 맞춰서 깔면됨

 

예를들어 쿠다 9.2를 깔기 위해 위같은 명령어에 torch 1.2를 깔고 싶으면 

conda install pytorch=1.2 torchvision cudatoolkit=9.2 -c pytorch

와 같이 깔아주면됨.

 

 

 

 

==> cuda와 torch를 세트로 깔아주면 좋을듯 

https://pytorch.org/get-started/previous-versions/

PyTorch

 

 

cuda 버전만 바꾸는법

 

conda install -c conda cudatoolkit=9.2

conda install -c conda cudnn=7