VS Code에서 Colab 연결하는 방법

Google Drive 마운트

왼쪽 탭에 있는 📁 아이콘을 눌러 Google Colab과 Drive 연결을 한다.

런타임 유형 GPU로 설정

[런타임] - [런타임 유형]에 들어가 하드웨어 가속기를 GPU로 바꿔준다.

ngrok의 authtoken 복사

ngrok 대시보드에 들어가 authtoken 값을 복사한다.

ssh 연결 준비

Google Colab에 들어가 다음의 코드를 적고, NGROK_TOKEN에 authtoken 값을 넣고, PASSWORD는 자신이 원하는 값으로 설정한다.

(나중에 ssh 연결 후 확인을 위해 비밀번호를 입력하는 창이 뜬다.)

!pip install colab-ssh  # colab-ssh 설치

NGROK_TOKEN = '[토큰값]'  # ngrok 토큰
PASSWORD = '[비밀번호]'   # 비밀번호 설정

from colab_ssh import launch_ssh
from colab_ssh import launch_ssh_cloudflared,init_git_cloudflared
launch_ssh(NGROK_TOKEN, PASSWORD)
launch_ssh_cloudflared(password=PASSWORD)

cloudflared 설치 및 config 파일 생성

위의 코드를 실행하면 다음과 같은 창이 뜨는데 cloudflared를 설치하고 config 파일을 생성하여 호스트 정보를 붙여넣고 저장한다.

이 때 주의할 점은 절대경로는 /cloudflared.exe로 끝나야한다.

VS Code에 Remote - SSH 확장 프로그램 설치

Visual Studio Code로 들어가 Remote - SSH를 설치해준다.

Remote - SSH로 호스트랑 연결

Ctrl+Shift+P를 눌러서 Remote - SSH: Connect to Host를 실행시키고 Google Colab에 있는 Hostname을 복사해서 붙여넣는다.

운영체제 선택 및 비밀번호

ssh로 Colab과의 연결이 시작되면 몇 개의 팝업이 뜨는 데, 플랫폼 운영체제는 Linux를 선택하고 앞서 Google Colab에 설정한 비밀번호를 입력하면 Google Colab과 VS Code가 연결된다!