구글 Colab 에서 PyTorch 사용하기

12시간 무료 GPU :D !

Google Colab 이란?

머신러닝 교육과 연구를 돕기 위한 플랫폼으로 Jupyter/iPython 기반의 노트북입니다 (https://colab.research.google.com). 구글 아이디만 있으면 CPU, GPU (12시간 후 런타임이 재설정 됨), TPU를 무료로 사용할 수 있습니다.

Colab 의 스펙?

• Python 2.7과 Python 3.6 사용 가능
• 무료 GPU 사용 가능 (12시간 후 런타임이 재설정 . i.e. 모든 로컬 변수와 파일 등 모든 런타임의 상태가 삭제됨)
• Tensorflow, Scikit-learn, Matplotlib 등 많은 python library 등이 설치되어 있어서 첫줄부터 바로 import ~~ 할 수 있음 (pytorch 는 불포함)
• Google Docs와 비슷하게 다운로드 없이 다른 개발자들과 협업할 수 있음
• Bash Command 사용 가능 (e.g. !pip install ~)
• 구글 드라이브, GitHub 레포, GitHub Gist 등으로 저장 가능

Colab 에서 PyTorch 사용하기

PyTorch는 설치되어있지 않아서 매 런타임 마다 install command 를 실행해야합니다 (ㅜ_ㅠ)

0. 구글 가입 후 → Google Colab Notebook 열기 → 새 Python 2 / 3 노트

또는

구글 드라이브 이동 → 새로 만들기 → 더보기 → Colaboratory

1. Colab 을 열면 CPU 가 런타임 유형 디폴트로 되어있습니다.
   GPU 로 바꾸기 위해서는 상단 바에 있는 [런타임] 에서 → [런타임 유형 변경] → [하드웨어 가속기]를 GPU 로 바꿔줍니다.

2. PyTorch 사용하기 (새로고침 / 런타임을 재설정 할 때마다 실행시켜야 합니다)

!pip3 install http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl
!pip3 install torchvision

두 줄의 코드를 실행시키면 약 2분 정도 후에 로딩이 완료됩니다.

코드를 실행시킨 후 예시 이미지

다음 셀부터 바로 import 하셔서 사용할 수 있습니다.

import torch
import torch.nn as nn
import torchvision.datasets as dsets

+) GPU 사용 할 때:

use_cuda = True
if use_cuda and torch.cuda.is_available():
    net.cuda()

+) 다른 library importing 예시들: https://colab.research.google.com/notebooks/snippets/importing_libraries.ipynb#scrollTo=RHXKNvj8ROgq