Pytorch 설치

이 글은 2025.01.28 기준이다.

 

1. 파이썬 환경 구성

먼저 pyenv를 이용해 적절한 python을 설치해 준다.

pyenv install 3.12.4

 

설치된 파이썬 확인

pyenv versions

 

파이썬 가상환경을 만들어 준다.

pyenv virtualenv 3.12.4 my_env

 

가상환경 확인

pyenv virtualenvs

 

가상환경 실행

activate my_env

 

---

2. Pytorch 설치

Pytorch 버전은 2.5.1 이 최신이다.

 

MPS (Apple Silicon) 지원 설치 (Mac M1/M2) 또는 CPU 전용 설치

pip install torch torchvision torchaudio

 

Cuda를 사용하는 경우

예시: CUDA 11.8

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

 

예시: CUDA 11.7

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

 

pytorch 설치 확인

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # GPU 사용 가능 여부 확인