Fix issues with mmdetection docker image. (#3515)

This commit is contained in:
rdondera-microsoft 2024-10-22 10:24:34 -07:00 коммит произвёл GitHub
Родитель 105e47deda
Коммит eb54a321f1
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
1 изменённых файлов: 5 добавлений и 0 удалений

Просмотреть файл

@ -9,9 +9,14 @@ COPY requirements.txt .
RUN pip install -r requirements.txt --no-cache-dir
# # # 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