Update Dockerfile

Dockerfile

@@ -1,11 +1,10 @@
-# 1. Use NVIDIA's stable Ubuntu 22.04 image (Most reliable for L40S drivers)
+# 1. Use NVIDIA's stable Ubuntu 22.04 image
 FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
 # Prevent interactive prompts
 ENV DEBIAN_FRONTEND=noninteractive
 
-# 2. Install Python 3.12 and Add PPA for modern C++ libraries
-# This fixes the CXXABI_1.3.15 error on Ubuntu 22.04
+# 2. Install Python 3.12 and modern C++ libraries (fixes the CXXABI error natively)
 RUN apt-get update && apt-get install -y software-properties-common && \
     add-apt-repository ppa:ubuntu-toolchain-r/test -y && \
     add-apt-repository ppa:deadsnakes/ppa -y && \
@@ -24,7 +23,7 @@ RUN apt-get update && apt-get install -y software-properties-common && \
 RUN useradd -m -u 1000 user
 USER user
 
-# 4. Set Architecture to 8.9 (L40S) and Paths
+# 4. Set Architecture for L40S (8.9) and CUDA paths
 ENV HOME=/home/user \
     PATH=/home/user/venv/bin:/usr/local/cuda/bin:$PATH \
     LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH \
@@ -37,24 +36,35 @@ WORKDIR $HOME/app
 # 5. Set up Virtual Environment
 RUN python3.12 -m venv $HOME/venv
 
-# 6. Install PyTorch for CUDA 12.1
-RUN pip install --no-cache-dir --upgrade pip && \
+# 6. Install PyTorch & Compilation Tools
+RUN pip install --no-cache-dir --upgrade pip wheel setuptools ninja && \
     pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
 
-# 7. CRITICAL FIX: Update requirements on the fly
-# We replace spconv-cu118 with spconv-cu120 (required for L40S)
+# 7. THE L40S FIX: Remove incompatible pre-compiled wheels
 COPY --chown=user requirements.txt .
-RUN sed -i 's/spconv-cu118/spconv-cu120/g' requirements.txt && \
+
+# We use 'sed' to dynamically delete the lines containing JeffreyXiang's pre-built .whl files
+# so pip doesn't install the broken binaries, then we install the rest of the file safely.
+RUN sed -i '/JeffreyXiang.*\.whl/d' requirements.txt && \
     pip install --no-cache-dir -r requirements.txt
 
-# 8. Copy application files
+# 8. Compile the 3D CUDA extensions natively for your L40S
+# Note: Because it is building heavy CUDA kernels from scratch, this step will 
+# take roughly 10 to 15 minutes during the Space build. Let it run!
+RUN pip install --no-cache-dir flash-attn --no-build-isolation && \
+    pip install --no-cache-dir git+https://github.com/JeffreyXiang/CuMesh.git && \
+    pip install --no-cache-dir git+https://github.com/JeffreyXiang/FlexGEMM.git && \
+    pip install --no-cache-dir git+https://github.com/microsoft/TRELLIS.2.git#subdirectory=o-voxel && \
+    pip install --no-cache-dir git+https://github.com/NVlabs/nvdiffrast.git && \
+    pip install --no-cache-dir git+https://github.com/JeffreyXiang/nvdiffrec.git@renderutils
+
+# 9. Copy application files
 COPY --chown=user . .
 
-# 9. App settings
+# 10. App settings
 EXPOSE 7860
 ENV GRADIO_SERVER_NAME="0.0.0.0"
 ENV GRADIO_SERVER_PORT="7860"
-# Since you have 48GB, we disable LOW_VRAM and optimize allocation
 ENV PYTORCH_CUDA_ALLOC_CONF="expandable_segments:True"
 
 # Run the app