Optimized Docker build to fix OOM errors

Dockerfile

@@ -1,5 +1,5 @@
 # Final stage Dockerfile - optimized for HuggingFace
-# Uses runtime base image (not devel) with minimal dependencies
+# Uses devel base image (includes CUDA compiler tools)
 FROM gpue/foundationpose-base:latest
 
 # FoundationPose configuration
@@ -9,14 +9,12 @@ ENV USE_REAL_MODEL=true
 # Downgrade NumPy to 1.x for CUDA extension compatibility
 RUN pip install --no-cache-dir "numpy<2"
 
-# Install CUDA compiler and build tools temporarily
+# Install build tools temporarily (CUDA compiler already in devel base)
 RUN apt-get update && apt-get install -y --no-install-recommends \
     cmake \
     build-essential \
     ninja-build \
     libeigen3-dev \
-    cuda-compiler-11-8 \
-    cuda-nvcc-11-8 \
     && rm -rf /var/lib/apt/lists/*
 
 # Build FoundationPose C++ extensions (requires GPU)
@@ -24,8 +22,8 @@ WORKDIR /app/FoundationPose
 RUN cd bundlesdf/mycuda && pip install . --no-build-isolation
 RUN cd mycpp && python setup.py build_ext --inplace
 
-# Remove build tools to save space
-RUN apt-get purge -y cmake build-essential ninja-build libeigen3-dev cuda-compiler-11-8 cuda-nvcc-11-8 && \
+# Remove build tools to save space (keep CUDA compiler)
+RUN apt-get purge -y cmake ninja-build libeigen3-dev && \
     apt-get autoremove -y && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*

Dockerfile.base

@@ -1,5 +1,5 @@
-# Minimal base image - optimized for size and build speed
-FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
+# Base image with CUDA compiler tools (needed for C++ extensions)
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV CUDA_HOME=/usr/local/cuda