Slim inference image with inference-only environment

Dockerfile

@@ -4,19 +4,18 @@ 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
 # for all subsequent RUN and CMD instructions.
 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 . .

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