[인공지능][AI] Pytorch(파이토치) 필수 라이브러리

Pytorch는 데이터 과학과 머신 러닝 작업을 위한 강력한 라이브러리로, 텐서 연산과 자동 미분 기능을 제공합니다.

 

Colab에 Pytorch가 이미 설치되어 있으므로 바로 import 할 수 있습니다.

Colab이 아닌 Local 환경에서 Pytorch를 이용하신다면 패키지 설치가 필요합니다.

기본적인 PyTorch 사용을 위해 다음 모듈들을 import 하는 것이 일반적입니다.

 

1. torch - 주요 PyTorch 라이브러리입니다.
2. torch.nn - 신경망 구축에 필요한 모듈, 함수, 클래스 등을 제공합니다.
3. torch.optim - 다양한 최적화 알고리즘을 제공합니다.
4. torch.utils.data - 데이터 로딩 및 전처리를 위한 유틸리티를 제공합니다.
5. torchvision - 이미지 데이터를 다루기 위한 도구 및 사전 학습된 모델을 제공합니다.

 

다음은 이러한 모듈들을 import하는 방법을 보여주는 예제 코드입니다:

import torch
import torch.nn as nn
import torch.optim as optim
import torch.utils.data as data
import torchvision
import torchvision.transforms as transforms