[IDE] Google Colab을 위한 팁: 세션 유지, GPU/TPU 할당, Google Drive 연동, Package Updata 반영

💡 본 문서는 'Google Colab을 위한 팁'에 대해 정리해놓은 글입니다.
Google Colab을 사용하며 필요한 기능에 대해 정리하였으니 참고하시기 바랍니다.

1. 세션 유지

colab은 가만히 놔두면 세션이 끊겨버리기 마련이기에, 간단히 colab 세션 유지하는 코드를 공유합니다 :)

아래 코드를 Chrome창 console command에 입력하면, 우측 상단에 리소스 탭(디스크, RAM 사용량)을 계속 눌려주는 것과 동일한 효과를 주게 되어 세션이 유지됩니다. (console은 F12를 누르면 확인하실 수 있습니다.)

function ConnectButton(){
    console.log("Connect pushed"); 
    document.querySelector("#top-toolbar > colab-connect-button").shadowRoot.querySelector("#connect").click() 
}
setInterval(ConnectButton,60000);

2. GPU/TPU 할당

2.1 하드웨어 가속기 할당

Colab에서 런타임 > 런타임 유형 변경 > 하드웨어 가속기에서 'GPU'/'TPU' 선택

2.2 하드웨어 가속기 설정 확인

TPU 사용에 있어 추가적인 팁을 얻고 싶다면 [Wikidocs] 18-01 코랩(Colab)에서 TPU 사용하기 를 참조하시기 바랍니다.

3. Google Drive 연동하기

google drive에 연동하여 드라이브에 있는 파일을 사용할 수 있습니다. 왼쪽 상단의 드라이브 버튼을 누르면 다음과 같은 코드가 생성되며, 이를 실행할경우 구글 드라이브와 연동됩니다.

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

4. Package Updata 반영

새로운 버전의 모듈을 설치했으면, runtime을 재시작해주어야 합니다.

Runtime(런타임) > Restart runtime(세션 다시 시작)

runtime을 restart할 경우 local variable 같은 정보는 사라지지만, 현재 환경 정보 (내가 설치한 모듈 버전 등) 은 갱신되게 됩니다.

참고

• [Blog] Google Colab session 유지: https://modernflow.tistory.com/12
• [Wikidocs] 18-01 코랩(Colab)에서 TPU 사용하기: https://wikidocs.net/119990
• [stackoverflow] need to restart runtime before import an installed package in colab: https://stackoverflow.com/questions/57831187/need-to-restart-runtime-before-import-an-installed-package-in-colab