Remove system GDAL packages and pin flash-attn to prevent build conflicts

- Remove libgdal-dev from apt-get to prevent conflict with compiled GDAL 3.11.5
- Add GDAL version verification to ensure correct version is available
- Pin flash-attn to 2.8.2 to match pre-built wheel on jetson-ai-lab.io

Fixes GDAL version detection issue where rasterio was finding system GDAL 3.4.1 instead of compiled 3.11.5, and flash-attn build failures when uv tried to compile 2.8.3 from source.

Author: alexnorell

docker/dockerfiles/Dockerfile.onnx.jetson.6.2.0

@@ -34,7 +34,9 @@ RUN apt-get update -y && \
     libwebp-dev \
     libzstd-dev \
     liblzma-dev \
-    && rm -rf /var/lib/apt/lists/*
+    && \
+    apt-get remove -y libgdal-dev gdal-bin libgdal30 2>/dev/null || true && \
+    rm -rf /var/lib/apt/lists/*
 
 # Compile GDAL from source to get version >= 3.5 for rasterio 1.4.0 compatibility
 RUN wget https://github.com/OSGeo/gdal/releases/download/v3.11.5/gdal-3.11.5.tar.gz && \
@@ -61,6 +63,10 @@ ENV GDAL_CONFIG=/usr/local/bin/gdal-config \
     LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH \
     PATH=/usr/local/bin:$PATH
 
+# Verify GDAL installation
+RUN gdal-config --version && \
+    test "$(gdal-config --version | cut -d. -f1,2)" = "3.11" || (echo "GDAL version mismatch!" && exit 1)
+
 RUN wget -q https://github.com/Kitware/CMake/releases/download/v3.30.5/cmake-3.30.5-linux-aarch64.sh && \
     chmod +x cmake-3.30.5-linux-aarch64.sh && \
     ./cmake-3.30.5-linux-aarch64.sh --prefix=/usr/local --skip-license && \

requirements/requirements.sam.txt

@@ -4,4 +4,4 @@ rasterio~=1.4.0
 pycocotools>=2.0.10
 torch>=2.8.0
 torchvision>=0.23.0
-flash-attn>=2.8.2
+flash-attn==2.8.2

requirements/requirements.transformers.txt

@@ -1,6 +1,6 @@
 torch>=2.8.0
 torchvision>=0.23.0
-flash-attn>=2.8.2
+flash-attn==2.8.2
 transformers>=4.53.3,<4.57.0
 timm~=1.0.0
 #accelerate>=0.32,<1.0.0