Colab에서 제공하는 패키지들 외에 다른 패키지를 사용하기 위해서는 pip install 명령어를 통해 직접 패키지를 설치해야 한다. 이때 새로운 세션을 시작하거나 런타임을 초기화하는 경우 패키지를 재설치해줘야하는 번거로움이 존재하는데, 이때 심폴릭 링크를 이용해 영구적으로 패키지를 설치할 수 있다.
import os, sys
from google.colab import drive
drive.mount('/content/drive')
my_path = '/content/package'
save_path = '/content/drive/MyDrive/Colab Notebooks/SD_packages'
os.symlink(save_path, my_path)
sys.path.insert(0, my_path)
구글 드라이브를 마운트하고 save_path에 패키지가 저장될 경로를 입력한다.
패키지가 설치되는 경로와 저장할 경로를 os.symlink를 통해 연결하고 아래처럼 --target 옵션을 통해 패키지를 영구적으로 설지할 수 있다.
!pip install --target=$my_path diffusers
!pip install --target=$my_path PIL
!pip install --target=$my_path einops
!pip install --target=$my_path insightface
!pip install --target=$my_path onnxruntime
그 후 다른 세션을 열 때나 재시작할 때 아래의 코드를 입력해주면 미리 설치해둔 패키지를 활용할 수 있다.
import os, sys
from google.colab import drive
drive.mount('/content/drive')
my_path = '/content/package'
save_path = '/content/drive/MyDrive/Colab Notebooks/SD_packages'
os.symlink(save_path, my_path)
sys.path.insert(0, my_path)
[Tip] Colab에 영구적으로 패키지 설치하기! - DACON
[Tip] Colab에 영구적으로 패키지 설치하기!
dacon.io
'메모 > Google Colab' 카테고리의 다른 글
[Colab] 마크다운으로 이미지 첨부 (0) | 2024.02.20 |
---|