[ETC] colab GPU vscode로 연결하기

colab gpu를 vscode로 연결해서 사용하는 방법을 리뷰하고자 한다.

remote ssh extension 설치

먼저 vscode에서 remote ssh extension을 설치한다.

cloudflare 설치

cloudflare이라는 프로그램을 설치한다.

설치 사이트

colab cloudflared 설치

colab에서 아래 코드를 실행하여 cloudflared를 설치해준다.

!pip install colab-ssh --upgrade

from colab_ssh import launch_ssh_cloudflared, init_git_cloudflared
launch_ssh_cloudflared(password='0000')

이를 실행하면 아래 화면이 뜰 것이다.

여기서 아래 하단에 ssh terminal에서의 명령 코드와 vscode에서의 remote ssh 코드가 있다. 이는 colab과 vscode를 연결해주는 방법 두 가지를 설명해주는 것인데, 두 가지 방법 다 가능하다.

ssh terminal

ssh terminal을 linux에서 실행하여 연결시킬 수 있지만, 우리는 vscode로 연결할 것이다. 먼저 vscode에서 원격 탐색기를 들어간다.

그런 다음 ssh targets를 선택해준다.

그리고 아래 +인 Add New를 선택하면 입력하는 창이 하나 뜬다. 거기에 코랩에 있던 좌하단 부분의 코드를 기입해준다.

그리고 enter를 치면 select ssh configuration 이라고 뜬다. 여기서 .ssh\config 를 선택하고 enter를 다시 치면 ssh targets 부분에 추가된 것을 볼 수 있다.

config 파일을 보면 host가 추가된 것을 확인할 수 있다. 추가되었으면 해당 ssh를 새 창에서 열면 사용이 가능하다. config에서 해당 host를 삭제하면 ssh targets에서 지워진다.

vscode remote ssh

1. 먼저 config 파일을 수정한다.

ctrl+shift+p 단축키를 누르고, open SSH configuration file을 선택한다. 그러면 config파일이 나오게 되고, 여기에 다음 코드를 추가한다.

Host *.trycloudflare.com
	HostName %h
	User root
	Port 22
	ProxyCommand <PUT_THE_ABSOLUTE_CLOUDFLARE_PATH_HERE> access ssh --hostname %h

이 부분을 추가하고, <>를 수정해줘야 한다. 위에서 cloudflare를 설치했다. 설치한 cloudflare 파일을 원하는 곳에 이동시키고, 실행시키면 config 파일이 생성된다. 그리고 그 파일의 경로를 <>에 넣어주면 된다.

Host *.trycloudflare.com
   HostName %h
   User root
   Port 22
   ProxyCommand C:\Users\dkssu\cloudflare\cloudflared-windows-amd64.exe access ssh --hostname %h

directory 가 아닌 file에 대해 경로를 넣어줘야 오류가 나지 않는다.

1. connect to host

위의 colab 애서 우하단의 코드를 복사한 후에 vscode에서 ctrl+shift+P 단축키를 눌러 remote SSH: connect to host 를 선택한다. 그리고 복사한 것을 넣어주고, enter를 누른다. 그러면 새 창이 나오고, 운영체제를 선택하라고 할 것이다. 그러면 linux를 선택하고, continue를 선택 후 아까 설정했던 password를 기입한다. 그러면 연결이 완료되고, 우측 탭에서 맨 위의 부분을 클릭해서 open folder를 하여 사용할 부분을 선택한다.

최종 화면

여기서 최종 확인을 위해 터미널에 nvidia-smi를 쳐본다.