[Colab/ngrok/vscode] Colab GPU - VSCode SSH 연동하여 local vscode에서 GPU 사용하기

출처

https://youtu.be/oAKxxLy-G5g

colab-vscode 연동

기본 개념

1. Colab file

https://colab.research.google.com/drive/1YS85pzYLSwVrS9Tk6BuDo-nSbLHUIyIz#scrollTo=OyUM6ddvvO8a

Google Colaboratory Notebook

GPU 선택

2. ngrok 계정 및 접속 코드

http://www.ngrok.com 가입한다.

https://dashboard.ngrok.com/get-started/setup 가입 후 뜨는 창에서 ngrok 접속코드 복사!

2번 ngrok 접속코드 복사한다.

3. vscode extension 설치
remote-ssh

remote-ssh 설치

4.VSCode - SSH 연결

command(ctrl) + Shift + P   ->  Remote-ssh  검색

command + shift + p -> remote ssh

Add News SSH Host 선택

from colab_ssh import launch_sshlaunch_ssh(NGROK_TOKEN,PASSWORD) 출력값

ssh <User>@<HostName> -p <Port>

오른쪽 하단 <Connect> 클릭

Colab 초반에 설정한 비번 변수

위에서 정한 비밀번호 입력

^(ctrl) + Shift + `(~키) -> 터미널 열기

nvidia-smi 입력

좌측 폴더 열기 /root/ 경로 선택 후 OK선택

비밀번호 입력

새파일 생성 후 test.py로 파일명 변경

GPU를 쓰는 tensorflow 라이브러리 버전확인

import tensorflow as tf
print(tf.__version__)