RTX 3090, 3080을 위한 pytorch 버전

pytorch에서 RTX3090 또는 RTX3080 을 이용한 가속을 사용하려고 할때, 그냥 설치해서 사용하면 GPU를 인식하지 못한다. 이때 나는 두 가지 버전 중 하나를 골라서 설치를 진행한다.

 

1. torch version == 1.7.1

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

 

2. torch version == 1.8.1

pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

 

위 두가지 버전을 기점으로 새로 생긴 함수나 지원안되는 함수가 많아서 두가지 중에서 알맞은 버전을 골라서 설치한다.