Composer 설치 — pip·conda·Docker

Composer 설치

Composer는 pip과 Anaconda로 손쉽게 설치할 수 있어요. 먼저 기본 설치부터 볼게요.

출처: https://docs.mosaicml.com/projects/composer/en/stable/getting_started/installation.html

pip / conda 설치

pip install mosaicml

conda install -c mosaicml mosaicml

선택적 의존성

일부 알고리즘·콜백·데이터셋·모델에 필요한 비핵심 의존성을 포함하는 설치 타깃이 있어요.

mosaicml[dev] — 테스트·문서 빌드용 개발 의존성
mosaicml[nlp] — NLP 모델·알고리즘 지원
mosaicml[wandb] — wandb 지원
mosaicml[comet_ml] — comet_ml 지원
mosaicml[neptune] — neptune 지원
mosaicml[tensorboard] — tensorboard 지원
mosaicml[streaming] — streaming 지원
mosaicml[mlflow] — mlflow 지원
mosaicml[oci] — oci 지원
mosaicml[onnx] — onnx 지원
mosaicml[coco] — coco 지원
mosaicml[libcloud] — libcloud 지원
mosaicml[all] — 모든 선택 의존성

개발자 설치로는 저장소를 클론해 설치해요.

git clone https://github.com/mosaicml/composer.git
cd composer
pip install -e ".[all]"

이미지 데이터를 빠르게 로드하려면 Pillow-SIMD 설치를 권장해요. vanilla pillow를 먼저 제거해야 하고, Apple M1 Mac에서는 지원되지 않아요.

pip uninstall pillow && pip install pillow-simd

Docker

환경 구성을 단순화하기 위해 사전 빌드된 Docker 이미지를 제공해요. mosaicml/composer 이미지는 NLP·비전 모델용 모든 의존성이 미리 설치돼 있고, mosaicml/pytorch 이미지 패밀리 위에 구축돼요. 가볍게 설치하려면 MosaicML PyTorch 이미지를 쓰고 그 안에서 Composer를 직접 설치하는 걸 권장해요.

docker pull mosaicml/composer