본문 바로가기

머신러닝/오류 모음집

Colab 초기 설정 튜토리얼

Colab 사용하면서 초기 셋팅 관련한 방법들을 정리해 보았습니다.

 

1. google drive 접근

2. matplot에서 한글 깨짐 현상

3. GPU 사용

4. Tensorflow 버전 설정

 

1. google drive에 접근하기

먼저 빈 창에 pwd 명령어를 입력하면 기본으로 지정되어 있는 path가 나옵니다. 저의 경우는 아래와 같이 나타났습니다.

이후 /content 를 시작으로 접근하고 싶은 주소를 drive.mount() 안에 입력해 줍니다.

from google.colab import drive

drive.mount('/content/drive')

 

2. matplot에서 한글 깨짐 현상

 

Mac OS 로컬에서 주피터를 실행할 때와 마찬가지로, matplot에서 한글이 깨지는 현상이 나타납니다.

우선, 아래 명령어를 실행하여 한글 폰트를 설치합니다.

!sudo apt-get install -y fonts-nanum
!sudo fc-cache -fv
!rm ~/.cache/matplotlib -rf

이후, rc를 이용하여 위에서 설치한 폰트로 설정해주면 colab에서도 위의 오른쪽 이미지와 같이 한글 폰트를 볼 수 있습니다.

import matplotlib.pyplot as plt
from matplotlib import font_manager, rc
plt.rcParams['axes.unicode_minus'] = False
rc('font', family='NanumBarunGothic')

plt.plot([1,2,3],[1,2,3])
plt.title('한글 제목')
plt.show()

 

단, 새로운 노트북을 열 때는 폰트를 다시 설치해주어야 합니다.

 

 

 

3. GPU 사용

탭 → 런타임 →  런타임 유형 변경  하드웨어 가속기  GPU

 

 

4. Tensorflow 버전 설정

colab에서 tensorflow는 버전 1.x와 2.x 중에 설정할 수 있습니다. 구체적인 버전 설정을 하려고 하면 자동으로 저 둘 중 하나로 정해지게 됩니다. tensorflow가 gpu를 사용하는 지 다음 셀로 확인할 수 있습니다. 

%tensorflow_version 2.x

import tensorflow as tf
from tensorflow.keras import datasets, layers, models
device_name = tf.test.gpu_device_name()
if device_name != '/device:GPU:0':
	raise SystemError('GPU device not found')
print('Found GPU at: {}'.format(device_name))