5초면 끝나는 Colab에서 Google Drive에 파일 저장하기(Mount 보다 한 걸음 더 나가기)

요즘 GPU가 필요해서 colab을 애용하고 있다.
그런데, colab의 런타임을 아무리 늘려도 어쩔 수 없이 런타임(런타임 늘리는 법은 여기 참고)이 끊길 때마다 작업하던 파일이 싹 다 날라가 버리는 슬픈일이 발생해서 작업한 파일을 바로 Google Drive에 저장하는 법을 찾아보기로 했다.

방법은 아주 간단하다!

가장 먼저, 구글 드라이브에 폴더를 하나 만들어주자. 필자는 아래와 같은 폴더들이 구글 드라이브에 있다.

그 다음 colab notebook 을 하나 열어서 다음 코드를 돌려주자.

# This mounts your Google Drive to the Colab VM.
from google.colab import drive
drive.mount('/content/drive')

그러면 구글 드라이브에 접근할 수 있게 된다.
그런데, 여기서 끝나면 섭섭하지. 현재 Workspace를 구글드라이브 폴더로 지정해보자. 이러면 구글 드라이브 속 특정 폴더를 워크 스페이스로 활용할 수 있다.

FOLDERNAME = 'test'

FOLERNAME 변수에 구글 드라이브 폴더 명을 넣어주자. 예를 들어, test 폴더 안에 test2 폴더가 있다면 'test/test2' 라고 입력해주면 되겠다.

FOLDERNAME = 'test'
assert FOLDERNAME is not None, "[!] Enter the foldername."

import sys
sys.path.append('/content/drive/MyDrive/{}'.format(FOLDERNAME))

# Change dariectory to current folder
%cd /content/drive/MyDrive/$FOLDERNAME

그 다음 위 코드를 돌려주면! 팝업창이 여러개 뜰 것이다.
확인, 허용 해주자.

Mounted at /content/drive
/content/drive/MyDrive/test

코드의 아웃풋에 위와 같은 결과가 프린트되면 잘 된 것이다! 이제 colab의 왼쪽에 있는 메뉴 창의 가장 아래쪽 폴더 같이 생긴 아이콘을 클릭해주자.

그럼 마운트가 잘 된 모습을 볼 수 있다.

그럼 workspace 가 잘 설정되어 있는지 확인해 보자. test 폴더 아래는 다음과 같은 파일이 있다.

현재 디렉터리 안에 있는 모든 파일을 보여주는 명령어 ls 를 돌려보자.

!ls

를 colab 상에서 돌려주면 된다.

아주 잘되고 있음을 볼 수 있다.

아주 유용하니 아래 코드를 복사해놓고 그때 그때 FOLERNAME 만 바꿔주면 된다.

# This mounts your Google Drive to the Colab VM.
from google.colab import drive
drive.mount('/content/drive')

FOLDERNAME = 'test'
assert FOLDERNAME is not None, "[!] Enter the foldername."

import sys
sys.path.append('/content/drive/MyDrive/{}'.format(FOLDERNAME))

# Change dariectory to current folder
%cd /content/drive/MyDrive/$FOLDERNAME

그럼 이만!