본문 바로가기
클라우드/colab

Colab을 이용해 인공지능을 개발해보자!-1편: Colab이란?

by kimjunhee9339 2024. 5. 15.

Colab은 구글에서 만든 클라우드 기반 주피터 노트북 웹 서비스이다. 

colab icon

Colab을 통해 pytorch나 tensorflow 기반의 인공지능을 제작할 수 있다.

주로 Colab에서 인공지능을 학습시키고 모델을 저장한 후에 aws ec2같은 데를 활용해 모델을 배포하게 된다.

필자는 aws의 sagemaker보다는 Colab을 추천한다. (sagemaker은 잘못 쓰면 비용 폭탄이 청구되기 때문이다.)

 

Colab 사용법:

1. https://colab.research.google.com/?hl=ko 에 접속한다.

2. 구글 로그인을 한다. 

3. 새노트 열기

4. 파이썬 코드 작성

다음과 같이 Colab을 사용할 수 있다.

 

파이썬 코드 작성 뿐만 아니라 설명 글 같은 것도 작성할 수 있다.

참고로 구글 드라이브의 파일들을 불러올 수 있는데 다음과 같다.

from google.colab import drive
drive.mount('/content/drive')

 

구글 드라이브를 업데이트하고 파일 모양 버튼을 클릭 후 새로고침을 누르면 파일이 업데이트된다.

필자는 주로 try catch문을 사용하여 error가 난다면 (해당 파일이 존재하지 않는다면) 구글 드라이브를 불러오고

error가 나지 않는 다면 구글 드라이브가 이미 연동되어 있으니 구글 드리아브를 연동하는 코드를 다시 실행하지 않는 방향으로 코드를 짜는데 다음과 같이 짤 수 있다.

try:
	'''
    내용...
    '''
except:
    from google.colab import drive
    drive.mount('/content/drive')
    '''
    내용...
    '''

 

 

요금제는 무료, pro, pro+, enterprise 가 있는데 자세한 것은 다음 링크를 통해 들어가면 확인 할 수 있다.

https://colab.research.google.com/signup?utm_source=footer&utm_medium=link&utm_campaign=footer_links

 

Colab 유료 서비스 가격 책정

 

colab.research.google.com

 

pytorch를 사용해 인공지능을 개발하는 것은 다음 편에 업로드하도록 하겠다.