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