[AI] Pytorch 개요 및 GPU 사용 여부 체크하기

PyTorch는 머신러닝 프레임워크 중 하나입니다.

Tensorflow도 머신러닝 프레임워크 중 하나죠.

PyTorch를 사용하면 GPU 연동을 통해 효율적으로 딥러닝 모델을 개발할 수 있습니다.

colab에서도 사용 가능합니다.

 

import torch

다음과 같이 import하여 사용할 수 있습니다.

 

GPU 사용 여부 체크하기

a = [[1,2],[3,4],[5,6]]
x = torch.tensor(a)
print(x.is_cuda)

x = x.cuda()
print(x.is_cuda)

x = x.cpu()
print(x.is_cuda)

x.cuda() 라는 명령어로 GPU로 옮기고, cpu()라는 명령으로  CPU로 옮길 수 있습니다.