[PyTorch] CUDA 11.2 + GTX 1080 Ti에 맞는 torch version 셋팅하기

 

MS에서 제공하는 Simple BaseLine을 실행시키려는데

 

가상환경 셋팅에 굉장히 많은 시간과 애를 먹었습니다..

 

저는 CUDA 11.2 버전 + GTX 1080 Ti 인데

 

해당 스펙에 맞는 버전을 찾아서 해결했습니다.

 

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

 

명령어를 보면 CUDA 11.0 버전으로 설치한 것을 알 수 있는데 11.2버전과도 호환 가능한 것으로 확인됩니다.