ARG VERSION=mmcv1.6_dtk26.04_v2.0_Hygon
FROM aicompute.tencentcloudcr.com/hyperdrive/hygon:${VERSION}

ARG PIP_INDEX_URL="https://pypi.tuna.tsinghua.edu.cn/simple"

RUN pip install -i "${PIP_INDEX_URL}" mmdet==2.28.0 mmsegmentation==0.14.1 \
    numba==0.56.4 numpy==1.23.5 && \
    sed 's/1.4.0/1.7.0/' -i /usr/local/lib/python3.10/dist-packages/mmseg/__init__.py

COPY BEVFormer /workspace/BEVFormer
COPY one_step_data.sh /workspace/BEVFormer/HyperDrive/tools/one_step_data.sh
COPY start_mmdet3d.sh /workspace/BEVFormer/tools/start_mmdet3d.sh

WORKDIR /workspace/BEVFormer
