From 478b02d1c1a050aa731170a16f784e578dc4040d Mon Sep 17 00:00:00 2001 From: Biplov Bhandari <4207677+biplovbhandari@users.noreply.github.com> Date: Fri, 28 Aug 2026 13:02:19 -0400 Subject: [PATCH] Slim inference image: split environment and update Dockerfile --- Dockerfile | 7 +++---- environment-inference.yaml | 14 ++++++++++++++ 2 files changed, 17 insertions(+), 4 deletions(-) create mode 100644 environment-inference.yaml diff --git a/Dockerfile b/Dockerfile index bf9d244..e1dae79 100644 --- a/Dockerfile +++ b/Dockerfile @@ -4,11 +4,10 @@ FROM mambaorg/micromamba WORKDIR /app # Copy the environment file -COPY --chown=$MAMBA_USER:$MAMBA_USER environment.yaml /tmp/env.yaml +COPY --chown=$MAMBA_USER:$MAMBA_USER environment-inference.yaml /tmp/env.yaml -# Install dependencies and awscli into the base environment +# Install inference-only dependencies (see environment-inference.yaml) RUN micromamba install -y -n base -f /tmp/env.yaml && \ - micromamba install -y -n base -c conda-forge awscli && \ micromamba clean --all --yes # This ARG forces the container to activate the 'base' environment @@ -16,7 +15,7 @@ RUN micromamba install -y -n base -f /tmp/env.yaml && \ ARG MAMBA_DOCKERFILE_ACTIVATE=1 # Fix for the libstdc++ issue (prevents "version `CXXABI_1.3.15' not found" errors) -ENV LD_LIBRARY_PATH="/opt/conda/lib:${LD_LIBRARY_PATH:-}" +ENV LD_LIBRARY_PATH="/opt/conda/lib" # Copy the source code into the container COPY --chown=$MAMBA_USER:$MAMBA_USER . . diff --git a/environment-inference.yaml b/environment-inference.yaml new file mode 100644 index 0000000..a5ba392 --- /dev/null +++ b/environment-inference.yaml @@ -0,0 +1,14 @@ +name: bridge-classify +channels: + - conda-forge +dependencies: + - python=3.11.* + - numpy=1.26.* + - python-pdal + - boto3 + - awscli + - pip + - pip: + - --extra-index-url https://download.pytorch.org/whl/cu126 + - torch==2.10.0+cu126 + - spconv-cu120==2.3.6