[Google Colaboratory] 코랩으로 GPU, TPU 사용법

개발자L

주제 :  코랩(Google Colab)으로 GPU, TPU 사용법

업데이트 일자 : 2022.02.01

---

1.  코랩(Colab) ?

구글 코랩(Google Colaboratory)는 Google Colaboratory의 줄임말로, 구글에서 무상으로 제공하는 클라우드 기반 Jupyter Notebook 개발환경입니다. 코랩의 가장 큰 장점은 자신의 컴퓨터가 GPU이 없거나 저사양인 경우에 런모드 유형 변경을 통해서 사용이 가능하며, 개인PC에 개발환경을 구축하는 수고를 덜어주어 개발에 시간을 투자 할수 있다는 점입니다. 최근, 머신러닝 및 딥러닝 학습에 많이 사용되고 있습니다.  
 - 개인PC의 개발환경에서 라이브러리 업데이트 및 설치시, 버젼이슈로 충돌되는 경우가 발생함
 - 코랩은 이러한 이슈로부터 비교적 자유로움

코랩 모식도(@개발자L)

그렇다면 코랩이 만능일까요?

'코랩 모식도'에서 볼 수 있듯이 코랩은 웹 브라우저(Web Browser)에서 동작합니다. 즉, 인터넷 연결이 되어있는 경우에만 작업이 가능합니다.  다시말해, 정부기관과 같이 보안상의 이유로 내부망을 사용하는 경우 또는 인터넷 연결에 어려움이 있는 곳에서는 사용에 제한이 있다는 단점을 가지고 있습니다. 또한, 특정 라이브러리가 필요하여 pip명령어로 설치하였다고 하더라도 웹브라우저를 닫는 경우, 라이브러리를 재설치 해야하는 번거로움이 있습니다.

---

2.  구글드라이브 연동 

 코랩을 사용하는 방법은 매우 간단합니다. 우선, 본인의 구글계정으로 로그인한뒤, 구글드라이브에서 오른쪽 클릭  >  더보기 > Google Colaboratory클릭 하시면 됩니다.

이후 자신의 구글드라이브와 연동하기 위해 아래의 코드를 입력하시고, 인증키를 기입하면 바로 사용이 가능합니다.

# 구글드라이브 연동
from google.colab import drive
drive.mount('/content/gdrive')

---

3.  런타임 유형 변경_GPU, TPU 

런타임 유형변경입니다. 상단의 "런타임" 클릭  >  "런타임 유형 변경" 클릭 > "GPU" 또는 "TPU" 택1하시면 됩니다.

구글드라이브에서 코랩 열기

런타임 유형 변경 방법

 

---

읽어볼거리. CPU, GPU, TPU란?  

지금까지 위에서 런모드 유형변경으로 변경하고자하는 GPU와 TPU란 무엇일까요?
타 블로그에 정리가 잘되어있는것들을 하단의 참조로 넣었습니다.

1. https://wooono.tistory.com/325
2. https://voidint.com/2020/10/14/cpu-gpu-tpu-npu/

[Hardware] CPU, GPU, TPU 간단 비교

CPU GPU TPU NPU 쉽게 개념 정리 합시다- voidint.com