Stats & AI tech blog - '일단 시도함'

[Python] Google Colab에서 로컬 GPU 연결하기 본문

Programming/Python

[Python] Google Colab에서 로컬 GPU 연결하기

justdoit ok? 2024. 4. 26. 14:28

Google Colab을 쓰다 GPU 사용 용량 제한 때문에 로컬 GPU에 연결해서 사용하기로 했다.

jupyter lab을 활용해서 연결을 해줄거기 때문에, Python과 jupyter notebook, jupyter lab을 설치가 필요하다.

 

 

1. Python  설치

https://www.python.org/downloads/

 

Download Python

The official home of the Python Programming Language

www.python.org

 

 

2. jupyter notebook 설치 

cmd > pip install jupyter notebook

 

 

3. jupyter lab 설치

cmd > pip install jupyter lab

        >  jupyter lab (설치 확인)

 

설치가 완료되었다면,

jupyter server가 colab의 연결을 허용할 수 있도록 config 파일 (설정파일)을 수정한다.

 

 

4. config 파일 수정하기 (colab 연결 액세스)

 

    4-1. config 파일 생성

    cmd > jupyter-lab --generate-config

 

    4-2. config 파일 수정

    파일을 생성하면 저장된 경로가 함께 출력되는데,

    해당 경로로 이동해서 jupyter_lab_config 파일을 열고, 아래 부분을 수정해준다.

 

    c.ServerApp.allow_origin = 'https://colab.research.google.com'

    c.ServerApp.port = 8888

    c.ServerApp.port_retries = 0

 

 

5. cmd 창에서 jupyer lab 실행, url 복사

config 파일을 수정했다면 다시 cmd 창에서 jupyter lab을 실행해주고, 이 때 출력되는 url을 복사해준다.

cmd > jupyter lab

 

6. colab에서 로컬 런타임에 연결

마지막으로 colab의.ipynb 코드를 열어 우측 상단의 연결 > 로컬 런타임에 연결을 클릭한다.

나타나는 창에 5.에서 복사한 url을 넣어주고 연결을 누르면 연결이 완료!

 

 

* 만약 연결에 오류가 난다면 jupyter lab console에서 아래 코드 실행

from jupyter_core import paths
print(paths.jupyter_config_path())

-> config파일을 가져오는 경로들이 출력됨.

출력된 경로에 수정한 config file 붙여넣은 후 다시 연결.