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

ARG COS_BASE_URL="https://haihub-model-1251001002.cos.ap-shanghai.myqcloud.com/HyperDrive/haiguang/hygon-bench/train"

SHELL ["/bin/bash", "-c"]

# Download Hyhal for compile
RUN aria2c --enable-color=false --show-console-readout=false -x 8 -s 8 -k 1M -d /tmp -o hyhal.tar.gz ${COS_BASE_URL}/common/hyhal.tar.gz && \
    mkdir -p /opt/hyhal && \
    tar -zxf /tmp/hyhal.tar.gz --strip-components=2 -C /opt/hyhal && \
    rm -rf /tmp/hyhal.tar.gz

COPY MapTRv2 /workspace/MapTRv2

RUN source /usr/local/bin/fastpt -C && \
    cd /workspace/MapTRv2/projects/mmdet3d_plugin/maptr/modules/ops/geometric_kernel_attn && \
    python setup.py build install && \
    pip uninstall -y mmdet3d && \
    pip install https://haihub-model-1251001002.cos.ap-shanghai.myqcloud.com/HyperDrive/haiguang/hygon-bench/train/autodrive/maptrv2/mmdet3d-0.17.2-cp310-cp310-linux_x86_64.whl && \
    source /usr/local/bin/fastpt -E || true # 编译成功也会返回1，忽略错误 @hygon

# 兼容 THPC 老模板
COPY build.sh /opt/build_maptrv2.sh

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

RUN rm -f /root/.bash_history

# Remove hyhal for runtime
RUN rm -rf /opt/hyhal

WORKDIR /workspace/MapTRv2
