v1.1.1

docker/Dockerfile

-ARG PYTORCH="1.8.1"
-ARG CUDA="10.2"
-ARG CUDNN="7"
+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk23.10-py38
+ENV DEBIAN_FRONTEND=noninteractive
+# RUN yum update && yum install -y git cmake wget build-essential
+RUN source /opt/dtk-23.10/env.sh
+# 安装pip相关依赖
+COPY requirements requirements
+COPY requirements.txt requirements.txt
+RUN pip3 install -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com -r requirements.txt
 
-FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
-
-ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0+PTX"
-ENV TORCH_NVCC_FLAGS="-Xfatbin -compress-all"
-ENV CMAKE_PREFIX_PATH="$(dirname $(which conda))/../"
-
-# To fix GPG key error when running apt-get update
-RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
-RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
-
-RUN apt-get update && apt-get install -y git ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 libgl1-mesa-glx\
-    && apt-get clean \
-    && rm -rf /var/lib/apt/lists/*
-
-# Install xtcocotools
-RUN pip install cython
-RUN pip install xtcocotools
-
-# Install MMEngine and MMCV
-RUN pip install openmim
-RUN mim install mmengine "mmcv>=2.0.0"
-
-# Install MMPose
-RUN conda clean --all
-RUN git clone https://github.com/open-mmlab/mmpose.git /mmpose
-WORKDIR /mmpose
-RUN git checkout main
-ENV FORCE_CUDA="1"
-RUN pip install -r requirements/build.txt
-RUN pip install --no-cache-dir -e .

docker/requirements.txt

+-r requirements/build.txt
+-r requirements/runtime.txt
+-r requirements/tests.txt
+-r requirements/optional.txt

docker/requirements/albu.txt

+albumentations>=0.3.2 --no-binary qudida,albumentations

docker/requirements/build.txt

+# These must be installed before building mmpose
+numpy
+torch>=1.8

docker/requirements/docs.txt

+docutils==0.16.0
+markdown
+myst-parser
+-e git+https://github.com/gaotongxiao/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme
+sphinx==4.5.0
+sphinx_copybutton
+sphinx_markdown_tables
+urllib3<2.0.0

docker/requirements/mminstall.txt

+mmcv>=2.0.0,<2.2.0
+mmdet>=3.0.0,<3.3.0
+mmengine>=0.4.0,<1.0.0

docker/requirements/optional.txt

+requests

docker/requirements/poseval.txt

+poseval@git+https://github.com/svenkreiss/poseval.git
+shapely==1.8.4

docker/requirements/readthedocs.txt

+mmcv>=2.0.0rc4
+mmengine>=0.6.0,<1.0.0
+munkres
+regex
+scipy
+titlecase
+torch>1.6
+torchvision
+xtcocotools>=1.13

docker/requirements/runtime.txt

+chumpy
+json_tricks
+matplotlib
+munkres
+numpy
+opencv-python
+pillow
+scipy
+# torchvision
+xtcocotools>=1.12
+mmdet
+mmengine

docker/requirements/tests.txt

+coverage
+flake8
+interrogate
+isort==4.3.21
+parameterized
+pytest
+pytest-runner
+xdoctest>=0.10.0
+yapf