Pytorch 버전 기록

Local

GPU : GTX 1660 Ti

 

PyTorch Install

$ pip3 install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio===0.10.1+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

 

Server

GPU : GTX 3090

 

PyTorch Install

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

CUDA & CUDNN

-- base
cuda_11.1.0_455.23.05_linux.run
cudnn-11.1-linux-x64-v8.0.5.39.tgz

-- edit
cuda_11.2.0_460.27.04_linux.run
cudnn-11.2-linux-x64-v8.1.0.77.tgz