각 CUDA 버전에 맞는 Pytorch 버전 확인하기

각 CUDA 버전마다 안정적으로 지원하는 Pytrorch 버전이 있다.

나중에 에러로 해매는 것 보다 미리 확인하고 까는 것을 추천한다 :)

 

CUDA 9.0 

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

 

CUDA 9.2

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

 

CUDA 10.0

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

 

CUDA 10.1

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

 

 

이런식으로 url을 수정하여 들어가보면 확인할 수 있음.