environments acft mmdetection image gpu - Azure/azureml-assets GitHub Wiki

acft-mmdetection-image-gpu

Overview

Environment used by MMDetection Image Finetune components

Version: 49

Tags

Preview

View in Studio: https://ml.azure.com/registries/azureml/environments/acft-mmdetection-image-gpu/version/49

Docker image: mcr.microsoft.com/azureml/curated/acft-mmdetection-image-gpu:49

Docker build context

Dockerfile

# PTCA image
FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2204-cu118-py310-torch222:biweekly.202412.3

USER root
RUN apt-get -y update && apt-get install -y libglib2.0-0

# Install required packages from pypi
COPY requirements.txt .
RUN pip install -r requirements.txt --no-cache-dir
RUN apt-get install -y expat=2.4.7-1ubuntu0.5

# # # Install mmdet
# Temporary workaround for https://github.com/open-mmlab/mim/issues/244
RUN pip install pip==24.0
# # Note that MMDet installs pycocotools
# Note: mmdet should be installed via mim to access the model zoo config folder.
RUN mim install mmdet==3.3.0
# Temporary workaround for https://github.com/open-mmlab/mmdetection/issues/11668 (when mmdet updated, remove lines below)
RUN mim install mmcv==2.2.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.2/index.html --no-cache-dir
RUN sed -i 's/2.2.0/2.3.0/' /opt/conda/envs/ptca/lib/python3.10/site-packages/mmdet/__init__.py

# Vulnerability Fix
RUN pip install gunicorn==22.0.0

# dummy number to change when needing to force rebuild without changing the definition: 2
⚠️ **GitHub.com Fallback** ⚠️