environments acft mmtracking video gpu

acft-mmtracking-video-gpu

Overview

Environment used by MMTracking Video Finetune components

Version: 26

Tags

Preview

View in Studio: https://ml.azure.com/registries/azureml/environments/acft-mmtracking-video-gpu/version/26

Docker image: mcr.microsoft.com/azureml/curated/acft-mmtracking-video-gpu:26

Docker build context

Dockerfile

# PTCA image
FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu117-py310-torch1131:biweekly.202405.1

USER root
RUN apt-get -y update

# Install required packages from pypi
COPY requirements.txt .
RUN pip install -r requirements.txt --no-cache-dir

# # # Install mmcv-full and mmdet
# RUN pip install openmim
RUN mim install mmtrack==0.14.0
RUN mim install mmcv-full==1.7.1
# # Note that MMDet installs pycocotools
# Note: mmdet should be installed via mim to access the model zoo config folder.
RUN mim install mmdet==2.28.2

# vulnerability fixes - this will be removed once we update to MMTracking's latest version
RUN pip install scipy==1.10.1
RUN pip install gunicorn==22.0.0

# It is recommended to upgrade to numpy version 1.23.5 to address the numpy.core.multiarray issue
RUN pip install numpy==1.23.5

# pin required version of yapf to avoid breaking changes in newer versions
# https://github.com/open-mmlab/mmdetection/issues/10962
RUN pip install yapf==0.40.1

# dummy number to change when needing to force rebuild without changing the definition: 1