Colab _ pytorch 실행 테스트

 

from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from torchvision.transforms import ToTensor, Lambda, Compose
import matplotlib.pyplot as plt
import numpy as np

https://gaussian37.github.io/dl-pytorch-pytorch-install/

PyTorch 설치 및 colab 사용 방법

• 먼저 colab 메뉴의 runtime → Change runtime type을 선택해서 GPU를 선택해 줍니다.
  • 딥러닝을 사용하는 것이니 GPU를 선택해 주어야 합니다.
   

• 아래 코드를 입력하여 PyTorch를 설치해줍니다.

!pip3 install torch
!pip3 install torchvision

 

다음으로 예제 코드를 실행한다.

from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from torchvision.transforms import ToTensor, Lambda, Compose
import matplotlib.pyplot as plt
import numpy as np

#MNIST Data down
#공개 데이터셋에서 학습 데이터를 내려받기
training_data = datasets.MNIST(
  root = 'data',
  train = True,
  download = True,
  transform = ToTensor(),
)

#공개데이터셋에서 테스트 데이터를 내려받아
test_data = datasets.MNIST(
    root='data',
    train=False,
    download=True,
    transform=ToTensor(),
)

batch_size = 64

#데이터로더생성
train_dataloader = DataLoader(training_data, batch_size= batch_size)
test_dataloader = DataLoader(test_data, batch_size = batch_size)

for x,y in test_dataloader : 
  print("Shape of x [N, C, H, W]: ", x.shape)
  print("Shape of y: ", y.shape, y.dtype)
  break