dockerfile changes for natten

Dockerfile

@@ -1,6 +1,6 @@
 FROM ubuntu:18.04
 
-ENV DEBIAN_FRONTEND noninteractive
+ENV DEBIAN_FRONTEND=noninteractive
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -11,12 +11,16 @@ RUN apt-get update && apt-get install -y \
     wget \
     curl \
     build-essential \
-    cmake \
-    libgl1-mesa-glx \
-    libglib2.0-0 \
+    libssl-dev \
+    libffi-dev \
+    libxml2-dev \
+    libxslt1-dev \
+    zlib1g-dev \
     libsm6 \
     libxext6 \
-    libxrender1 \
+    libxrender-dev \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
     libgomp1 \
     && rm -rf /var/lib/apt/lists/*
 
@@ -24,62 +28,54 @@ RUN apt-get update && apt-get install -y \
 RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.8 1 && \
     update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 1
 
+# Upgrade pip
+RUN python -m pip install --upgrade pip==22.3.1 setuptools==59.5.0 wheel
+
 # Create user
 RUN useradd -m -u 1000 user
 WORKDIR /app
 
-# Upgrade pip
-RUN python -m pip install --upgrade pip==21.3.1 setuptools==59.5.0 wheel
-
-# Install PyTorch CPU (specific version that works with detectron2)
+# Install PyTorch CPU first
 RUN pip install torch==1.9.0+cpu torchvision==0.10.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
 
-# Install numpy first (dependency for many packages)
-RUN pip install numpy==1.21.6
-
-# Build and install detectron2 from source for compatibility
-RUN pip install cython pybind11
-RUN git clone https://github.com/facebookresearch/detectron2.git && \
-    cd detectron2 && \
-    git checkout v0.6 && \
-    pip install -e . && \
-    cd .. && \
-    rm -rf detectron2/.git
-
 # Install core dependencies with compatible versions
 RUN pip install \
+    numpy==1.19.5 \
     Pillow==8.3.2 \
     opencv-python==4.5.3.56 \
+    cython==0.29.24
+
+# Install detectron2 CPU
+RUN pip install detectron2==0.6 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.9/index.html
+
+# Install compatible versions for Python 3.8
+RUN pip install \
+    gradio==3.1.7 \
+    huggingface_hub==0.8.1 \
     scipy==1.7.1 \
     scikit-image==0.18.3 \
     scikit-learn==1.0.2 \
-    h5py==3.1.0 \
-    shapely==1.7.1 \
-    imutils==0.5.4
-
-# Install ML dependencies with compatible versions
-RUN pip install \
     timm==0.4.12 \
     einops==0.3.2 \
     tqdm==4.62.3 \
-    pycocotools
-
-# Install compatible gradio and huggingface_hub
-RUN pip install \
-    gradio==3.1.7 \
-    huggingface_hub==0.10.1
-
-# Install other dependencies
-RUN pip install \
-    gdown==4.5.1 \
-    wget==3.2 \
-    ftfy==6.0.3 \
+    imutils==0.5.4 \
+    shapely==1.8.0 \
+    h5py==3.1.0 \
     regex==2021.11.10 \
+    ftfy==6.0.3 \
     inflect==5.3.0 \
-    submitit==1.4.1
+    gdown==4.2.0 \
+    wget==3.2
+
+# Optional dependencies (skip if causing issues)
+RUN pip install submitit==1.4.1 || echo "submitit failed"
+RUN pip install pytorch_lightning==1.5.10 || echo "pytorch_lightning failed"
+RUN pip install wandb==0.12.9 || echo "wandb failed"
+RUN pip install icecream==2.1.1 || echo "icecream failed"
 
-# Optional: pytorch_lightning (skip if not needed for faster builds)
-# RUN pip install pytorch_lightning==1.5.10
+# Try NATTEN
+RUN pip install natten==0.14.6 -f https://shi-labs.com/natten/wheels/cpu/torch1.9/index.html || \
+    echo "NATTEN installation failed"
 
 # Switch to user
 USER user
@@ -88,7 +84,7 @@ ENV HOME=/home/user PATH=/home/user/.local/bin:$PATH
 # Copy application
 COPY --chown=user:user . /app
 
-# CPU environment variables
+# CPU environment
 ENV CUDA_VISIBLE_DEVICES=""
 ENV FORCE_CUDA="0"
 ENV OMP_NUM_THREADS=4