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 mim and mmcv version

Browse Source
This commit is contained in:
triple-Mu 2022-11-10 11:18:56 +08:00
parent b6cbbdc212
commit f1d2d09338
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docker/CPU/Dockerfile
View File

@ -3,7 +3,7 @@ ARG PYTHON_VERSION=3.7
ARG TORCH_VERSION=1.10.0 ARG TORCH_VERSION=1.10.0
ARG TORCHVISION_VERSION=0.11.0 ARG TORCHVISION_VERSION=0.11.0
ARG ONNXRUNTIME_VERSION=1.8.1 ARG ONNXRUNTIME_VERSION=1.8.1
ARG MMCV_VERSION=">=2.0.0rc2,<2.1.0" ARG MMCV_VERSION=">=2.0.0rc2"
ARG MMENGINE_VERSION=">=0.3.0" ARG MMENGINE_VERSION=">=0.3.0"
USER root USER root
@ -69,7 +69,7 @@ RUN git clone https://github.com/Tencent/ncnn.git &&\
make -j$(nproc) &&\ make -j$(nproc) &&\
make install &&\ make install &&\
cd /root/workspace/ncnn/python &&\ cd /root/workspace/ncnn/python &&\
mim install -e . /opt/conda/bin/mim install -e .
ENV PATH="/root/workspace/ncnn/build/tools/quantize/:${PATH}" ENV PATH="/root/workspace/ncnn/build/tools/quantize/:${PATH}"
@ -88,7 +88,7 @@ RUN git clone https://github.com/open-mmlab/mmdeploy.git &&\
cmake -DMMDEPLOY_TARGET_BACKENDS=ort .. &&\ cmake -DMMDEPLOY_TARGET_BACKENDS=ort .. &&\
make -j$(nproc) &&\ make -j$(nproc) &&\
cd .. &&\ cd .. &&\
mim install -e . /opt/conda/bin/mim install -e .
### build SDK ### build SDK
ENV LD_LIBRARY_PATH="/root/workspace/mmdeploy/build/lib:/opt/intel/openvino/deployment_tools/ngraph/lib:/opt/intel/openvino/deployment_tools/inference_engine/lib/intel64:${LD_LIBRARY_PATH}" ENV LD_LIBRARY_PATH="/root/workspace/mmdeploy/build/lib:/opt/intel/openvino/deployment_tools/ngraph/lib:/opt/intel/openvino/deployment_tools/inference_engine/lib/intel64:${LD_LIBRARY_PATH}"

2
docker/GPU/Dockerfile
View File

@ -7,7 +7,7 @@ ARG TORCHVISION_VERSION=0.11.0
ARG ONNXRUNTIME_VERSION=1.8.1 ARG ONNXRUNTIME_VERSION=1.8.1
ARG PPLCV_VERSION=0.7.0 ARG PPLCV_VERSION=0.7.0
ENV FORCE_CUDA="1" ENV FORCE_CUDA="1"
ARG MMCV_VERSION=">=2.0.0rc2,<2.1.0" ARG MMCV_VERSION=">=2.0.0rc2"
ARG MMENGINE_VERSION=">=0.3.0" ARG MMENGINE_VERSION=">=0.3.0"
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive