머신러닝/오류 모음집

Colab 초기 설정 튜토리얼

amapro 2022. 6. 6. 10:30

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))