[파이썬] 파이토치 설치

 

설치 사이트

https://pytorch.org/get-started/locally/

Start Locally

 

cuda 버전 확인

 

파이참  터미널 

nvidia-smi

 

 

 

사이트에 원하는 항목 선택

conda 나 pip 원하는 항목 선택  

가장 아래 커맨드 복사하여 파이참 터미널에 입력

 

설치시작

 

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

 

CPU 버전

pip3 install torch torchvision torchaudio

 

 

설치 완료 확인

import torch # 텐서 생성 테스트 x = torch.rand(5, 3) print(x) # GPU 사용 가능 여부 확인 print(f"CUDA 사용 가능 여부: {torch.cuda.is_available()}")