[PyTorch] Pytorch 설치

[PyTorch] CUDA, cuDNN 설치 — zerogod 코코딩딩

[PyTorch] CUDA, cuDNN 설치

CUDA와 cuDNN은 위에서 설치

 

Start Locally | PyTorch

Start Locally

위 링크에 접속한다. pip를 이용한 설치를 진행할 예정이며 최소 python 3.9 이상부터 가능하다고 한다.

 

설치한 Toolkit에 맞게 설정해준다. Run this Command를 복사해 Command Prompt에서 실행해준다. 

 

정상적으로 설치가 되었는지 확인해보자. 

 

import torch
print(torch.__version__)  # PyTorch 버전 출력
print(torch.cuda.is_available())  # True: GPU 사용 가능
print(torch.cuda.get_device_name(0))  # GPU 이름 확인

print(torch.backends.cudnn.version())  # cuDNN 버전 출력

2.5.1+cu124
True
NVIDIA GeForce RTX 4050 Laptop GPU
90100

 

설정이 완료되었다.