install Pytorch and Cuda on Windows10, GeForce2080Ti

1. CUDA 설치 - CUDA v11.0.3

CUDA 11.0 공홈 설치 사이트

GeForce 2080Ti의 경우, CUDA 11.0 이상 설치해야 실행할 수 있음.
위 사이트 방문하여 설치함.
아래 명령어 사용하여 설치 확인 가능

$ nvcc --version

2. CUDA CNN 설치 - CuDNN v8.4.0

리눅스 기준 설치하지 않아도 가능했지만, 윈도우의 경우 다를 수 있어 설치함.
CUDA CNN 설치

zip 파일 내에 있는 모든 파일을 아래 경로에 모두 옮겨넣음.

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0

3. Python 설치 - Python 3.9.0

anaconda, miniconda 설치도 가능함.
단, CUDA v.11.0 이상은 Python v3.8 이상만 지원.

$ python --version

4. PyTorch 설치 - v1.7.1(+cu110)

pytorch 설치
위 사이트 참고하여서 CUDA 버전에 맞는 PyTorch 설치 완료함.

$ python
>> import torch
>> torch.__version__