[Pytorch/파이토치] CUDA Error: no kernel image is available for execution on the device

사용하고 있는 GPU : Geforce RTX3090 

서버에서 모델을 돌려보려고, 가상환경에 pytorch를 설치해서 돌려보았다.

역시.. 한 번에 되는게 없다.. 

이번에 GPU를 사용하게 되면서 모델을 돌리기 위해서 Torch를 사용하고 CUDA에 관한 에러를 많이 접했는데, 자신이 사용하고 있는 GPU에 맞게 Pytorch 버전을 설치해야한다고 한다.

아래와 같이 다시 설치를 해보았다.

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

RTX3090을 지원하는 Pytorch는 python3.8이상을 요구한다고 한다.

아래와 같이 python3.8에서 테스트한 결과

잘 해결된 것을 확인할 수 있다.