[PyTorch] CUDA 11.2 + RTX3090에 맞는 torch version 세팅하기

Causal HTP 코드를 돌리는데 가상환경 세팅에 조금 애를 먹었다.

원인은 CUDA 버전과 torch 버전이 안맞아서,,,, !

코드를 돌려도 model.cuda() 부분에 딜레이가 생기고 결국 epoch 진행이 되지 않았다. 

결국 11.2에 맞는 torch version을 다음과 같이 설치해 해결하였다. 

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

 

---

Reference 

https://sanglee325.github.io/environment/pytorch-cuda112/#problem