26 lines
1.1 KiB
Docker
26 lines
1.1 KiB
Docker
FROM openmmlab/mmdeploy:ubuntu20.04-cuda11.3
|
|
|
|
ARG MMDEPLOY_VERSION
|
|
|
|
ENV BACKUP_LD_LIBRARY_PATH=$LD_LIBRARY_PATH
|
|
ENV LD_LIBRARY_PATH=/usr/local/cuda/compat:$LD_LIBRARY_PATH
|
|
|
|
# build mmdeploy
|
|
RUN git clone --recursive -b $MMDEPLOY_VERSION --depth 1 https://github.com/open-mmlab/mmdeploy.git &&\
|
|
export Torch_DIR=$(python3 -c "import torch;print(torch.utils.cmake_prefix_path + '/Torch')") &&\
|
|
bash mmdeploy/.circleci/scripts/linux/build.sh "cpu;cuda" "ort;trt;ncnn;torchscript" \
|
|
-Dpplcv_DIR=${pplcv_DIR} \
|
|
-DTENSORRT_DIR=${TENSORRT_DIR} \
|
|
-DONNXRUNTIME_DIR=${ONNXRUNTIME_DIR} \
|
|
-Dncnn_DIR=${ncnn_DIR} \
|
|
-DTorch_DIR=${Torch_DIR} &&\
|
|
cd mmdeploy &&\
|
|
python3 -m pip install -U openmim pycuda &&\
|
|
python3 -m pip install -r requirements.txt &&\
|
|
python3 -m pip install -e .
|
|
|
|
ENV MMDeploy_DIR="/root/workspace/mmdeploy/build/install/lib/cmake/MMDeploy"
|
|
ENV LD_LIBRARY_PATH="/root/workspace/mmdeploy/build/lib:${BACKUP_LD_LIBRARY_PATH}"
|
|
ENV PATH="/root/workspace/mmdeploy/build/bin:${PATH}"
|
|
ENV PYTHONPATH="/root/workspace/mmdeploy/build/lib:${PYTHONPATH}"
|