Backend 3
PyTorch
PyTorch는 machine learning과 GPU 기반 연구 코드에서 자주 사용하는 tensor library입니다.
PyTorch를 쓸 때
설치 안내, 문서, ecosystem 정보를 확인해야 할 때는 PyTorch 공식 사이트를 사용합니다.
PyTorch 공식 사이트Neural network, GPU tensor operation, model training loop가 workflow에 포함될 때 PyTorch를 사용합니다.
Tensor와 automatic differentiation이 필요한 수치 실험에도 사용할 수 있습니다.
최소 예제
import torch
x = torch.tensor([0.0, 1.0, 2.0])
y = x ** 2
print(y)
작업 습관
CPU와 GPU 코드를 섞기 전에는 tensor가 어느 device에 있는지 확인합니다. Device mismatch는 초기에 자주 만나는 오류입니다.
print(x.device)