mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

Fix mmengine mmcv version

Browse Source
This commit is contained in:
triple-Mu 2022-11-04 15:18:02 +08:00
parent c74531de63
commit 9263c7f604
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docker/CPU/Dockerfile
View File

@ -3,7 +3,8 @@ ARG PYTHON_VERSION=3.7
ARG TORCH_VERSION=1.10.0
ARG TORCHVISION_VERSION=0.11.0
ARG ONNXRUNTIME_VERSION=1.8.1
ARG MMCV_VERSION
ARG MMCV_VERSION=">=2.0.0rc2,<2.1.0"
ARG MMENGINE_VERSION=">=0.3.0"
USER root
### change the system source for installing libs
@ -48,8 +49,7 @@ RUN /opt/conda/bin/pip install --no-cache-dir openmim
### pytorch mmcv onnxruntime and openvino
RUN mim install torch==${TORCH_VERSION}+cpu torchvision==${TORCHVISION_VERSION}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html \
&& if [ -z ${MMCV_VERSION} ] ; then MMCV="mmcv>=2.0.0rc2,<2.1.0" ; else MMCV="mmcv=="${MMCV_VERSION} ; fi && \
&& mim install --no-cache-dir ${MMCV} onnxruntime==${ONNXRUNTIME_VERSION} openvino-dev
&& mim install --no-cache-dir "mmcv"${MMCV_VERSION} onnxruntime==${ONNXRUNTIME_VERSION} openvino-dev mmengine${MMENGINE_VERSION}
ENV PATH /opt/conda/bin:$PATH

6
docker/GPU/Dockerfile
View File

@ -8,7 +8,8 @@ ARG ONNXRUNTIME_VERSION=1.8.1
ARG MMCV_VERSION=1.5.3
ARG PPLCV_VERSION=0.7.0
ENV FORCE_CUDA="1"
ARG MMCV_VERSION
ARG MMCV_VERSION=">=2.0.0rc2,<2.1.0"
ARG MMENGINE_VERSION=">=0.3.0"
ENV DEBIAN_FRONTEND=noninteractive
@ -40,8 +41,7 @@ RUN /opt/conda/bin/pip install --no-cache-dir openmim
### pytorch mmcv onnxruntime
RUN mim install torch==${TORCH_VERSION}+cpu torchvision==${TORCHVISION_VERSION}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html \
&& if [ -z ${MMCV_VERSION} ] ; then MMCV="mmcv>=2.0.0rc2,<2.1.0" ; else MMCV="mmcv=="${MMCV_VERSION} ; fi && \
&& mim install --no-cache-dir ${MMCV} onnxruntime-gpu==${ONNXRUNTIME_VERSION}
&& mim install --no-cache-dir "mmcv"${MMCV_VERSION} onnxruntime-gpu==${ONNXRUNTIME_VERSION} mmengine${MMENGINE_VERSION}
ENV PATH /opt/conda/bin:$PATH