[Python] 다른 폴더에 있는 패키지, 모듈 가져오기 (sys.path.insert)

오늘은 Google Colab에서 현재 작업 폴더가 아닌 다른 폴더에 있는 패키지나 모듈을 import하는 방법에 대해 알아보겠다.

우선 Colab과 Google Drive를 연동시켜주는 작업이 먼저 필요해요. 아래 코드를 실행하면 연동이 가능하다.

from google.colab import drive
drive.mount('/content/drive')

 

코드를 실행시킨 후 잠시 기다리면 화면 왼쪽 파일이 표시되는 부분에 drive 폴더가 생긴 걸 확인할 수 있다.

 

 

드라이브가 연동되었다면 이제 다른 폴더에 있는 패키지/모듈을 import 해보겠다.

 

"drive > Mydrive > PyTorch > Image Classification > HERBS" 안에 있는 timm 패키지를 import 하고 싶다고 할 때,

바로 import timm을 수행하게 되면  "ModuleNotFoundError: No module named 'timm'"이라는 에러 메시지가 뜨며 import에 실패하게 된다.

 

 

이런 경우, 먼저 sys.path.insert(0, 경로)를 통해 패키지가 저장된 경로를 추가해주고, import를 실행하면 정상적으로 import 되는 것을 확인할 수 있다.

import os, sys
sys.path.insert(0, '/content/drive/MyDrive/HERBS') 
import timm

 

 

추가로, timm 패키지 안에 포함된 특정 모듈만 가져오고 싶다면 아래 식을 통해 가져올 수 있다.

from timm import 모듈명