[Setup] Window | Anaconda | Pytorch | CUDA | CUDNN

CUDA - torch 버전 차이로 삽질

뭐하려고 했는지 모르겠

---

 

사전 확인: GPU 체크

PC에 장착된 GPU 모델 확인

 

 

❍ NVIDIA GeForce GTX 1080 Ti

 

 

❍ CUDA Version: 12.0: 내 그래픽 카드에 설치할 수 있는 버전

 

---

Anaconda 설치

 

❍ python 3.9.13

❍ conda 23.5.2

 

---

CUDA 설치

 

 

nvcc -V로 버전 확인 가능

❍ CUDA: V11.2.67

 

 

CUDNN Version Check (CUDNN 8 이후)

C: > Program Files > NVIDIA GPU Computing Toolkit > CUDA > v11.2(자신 버전) > include > cudnn_version.h

 

MAJOR - MINOR - PATCHLEVEL 순으로 버전 = 8.1.0

 

---

 

Torch 설치

 

공식 홈페이지

PyTorch

 

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

 

import torch

print(torch.cuda_is_available()) # True