[PyTorch] 개요 및 설치 가이드

소개

  PyTorch는 메타(Facebook)에서 개발한 오픈소스 딥러닝 라이브러리로, 직관적이며 유연한 텐서 연산과 GPU 가속, 자동 미분을 지원하여 빠른 프로토타이핑 및 연구 개발에 널리 사용됩니다.

 

설치 방법

# 기본 설치 (CPU 전용)
pip install torch torchvision torchaudio

# GPU 지원 설치 (CUDA 12.1 예시)
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

 

설치 확인

import torch
print(torch.__version__, torch.cuda.is_available())

 

관련 내용

  PyTorch 공식 홈페이지

  PyTorch 자주 쓰이는 함수 모음

[PyTorch] 함수 목차