Colab: 내장 패키지 확인 및 업그레이드 방법 소개

안녕하세요? 이번에는 Google Colab에서 내장 패키지 목록을 확인하고, 패키지 버전을 업그레이드 하는 방법을 간략히 소개해 보겠습니다. 이번 글은 인도네시아 PT Mobility Doctor Indonesia에서 개발자로 활동하시는 배진익 님이 알려주신 팁을 재정리한 것입니다.

 

Google Colab

Google Colab(구글 코랩)은 Google 리서치팀에서 개발한 제품입니다. Colab은 호스팅된 Jupyter Notebook 서비스로, 누구나 브라우저를 통해 임의의 Python 코드를 작성하고 실행할 수 있으며 GPU를 포함한 컴퓨팅 리소스를 무료로 사용할 수 있습니다.

구글 코랩(Google Colab)으로 머신러닝 프로그램 돌리기 : github 예제 有

Colab은 tensorflow, torch, keras, GDAL, geopy, seaborn, matplot 등 다수의 내장 패키지를 제공하고 있습니다. 아래 명령어를 통해 Colab의 내장 패키지 목록과 버전을 확인하실 수 있습니다.

# Colab 내장 패키지 확인
!pip list

output:
Package                       Version
----------------------------- ----------------------
absl-py                       1.3.0
aeppl                         0.0.33
aesara                        2.7.9
aiohttp                       3.8.3
aiosignal                     1.3.1

목록에서 Matplotlib 패키지 버전을 확인해 보겠습니다. Colab은 현재 Matplotlib 3.2.2 버전을 제공하고 있는에요, 이로 인해 실습과정에서 일부 오류를 경험하실 수 있습니다. 따라서 패키지 업그레이드가 필요합니다.

# Matplotlib 버전 확인
!pip list | grep matplotlib

output:
matplotlib                    3.2.2
matplotlib-venn               0.11.7

배진익 님이 공유해주신 matplotlib 버전 확인 및 업그레이드 방법입니다. packaging 패키지로 해당 패키지 기준 버전 미만인지 체크한 후, 필요 시 업그레이드하는 방식입니다. 프로세스를 종료하는 os.kill() 메소드가 추가되어 있어 런타임 종료 후 현재 셀을 다시 실행해주시면 됩니다. 유익한 코드죠?!

# matplotlib 버전 확인 및 업그레이드
!pip install packaging
from packaging import version

import matplotlib
print("current version: " + matplotlib.__version__)

if(version.parse(matplotlib.__version__) < version.parse("3.6.0")):
    print("업그레이드 시작")
    !pip install --upgrade matplotlib -q
    print("업그레이드 완료 후 런타임 종료. 현재 셀을 다시 실행해주세요.")
    import os
    os.kill(os.getpid(), 9) # 런타임 종료

Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/
Requirement already satisfied: packaging in /usr/local/lib/python3.8/dist-packages (21.3)
Requirement already satisfied: pyparsing!=3.0.5,>=2.0.2 in /usr/local/lib/python3.8/dist-packages (from packaging) (3.0.9)
current version: 3.2.2
업그레이드 시작
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 9.4/9.4 MB 63.2 MB/s eta 0:00:00
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 965.4/965.4 KB 60.6 MB/s eta 0:00:00
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 300.0/300.0 KB 28.6 MB/s eta 0:00:00

셀을 다시 실행해 보면 Matplotlib이 최신 버전(3.6.3)으로 업그레이드된 것을 확인하실 수 있습니다.

output:
Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/
Requirement already satisfied: packaging in /usr/local/lib/python3.8/dist-packages (21.3)
Requirement already satisfied: pyparsing!=3.0.5,>=2.0.2 in /usr/local/lib/python3.8/dist-packages (from packaging) (3.0.9)
current version: 3.6.3

Colab_내장_패키지_확인_및_업그레이드.ipynb