Colab VScode 연동

회사에서 새로운 GPU 환경을 구성을 계획하고 있는데, 그때까지 Colab Pro Plus 에 대한 사용경비 청구를 승인해줘서 두달 간 사용해보았다.

업무 개인 PC에서는 rtx2070 8GB가 장착되어 있는데, ( 이젠 OOM 만 보면 PTSD가.. ) Colab Pro Plus 에서는 항상 P100 16GB 을 할당 받아 사용할 수 있었다. ( V100도 사용해본 분이 있다고 들었는데, 저는 한번도 안뽑혔다는... )

그 과정에서 내가 사랑하는 VScode 에 Colab을 연동해서 사용한 setting 과정을 공유하고자 한다.

 

cloudflared 다운로드

local 운영체제에 적합한 cloudflared.exe 를 다운로드한다

https://developers.cloudflare.com/cloudflare-one/connections/connect-apps/install-and-setup/installation

 

Colab

1. Colab 노트북 파일을 하나 생성한다

2. ssh host 및 pw 설정: Colab에서 아래 코드블럭 실행

3. host 복사: 결과창의 우측 하단에 나오는 “copy” 클릭

!pip install colab-ssh --upgrade

from colab_ssh import launch_ssh_cloudflared, init_git_cloudflared
launch_ssh_cloudflared(password='test') # password는 임의로 설정

 

VScode

4. ssh접속 설정 : .ssh 디렉토리 내 config.json 에 아래 내용 추가

• ProxyCommand에 자신의 local 내 cloudflared.exe PATH입력

Host *.trycloudflare.com
    HostName %h
    User root
    Port 22
    ProxyCommand C:\\Users\\user\\Desktop\\cloudflared.exe access ssh --hostname %h

 

5. 명령창(Ctrl + Shift + P) 에서 “Remote-SSH: Connect to Host에서 Paste 실행 및 접속

- “Linux”, “Continue” 를 차례로 클릭, PW 입력

 

<Reference>

https://data-newbie.tistory.com/699