Fixes for python 3.10

Dockerfile

@@ -1,14 +1,20 @@
-FROM nvidia/cuda:12.4.1-cudnn8-devel-ubuntu22.04
+FROM python:3.10-slim
 
-# Remove Python 3.9 that comes with base image
-RUN apt-get update && apt-get remove -y python3.9 python3.9-minimal python3.9-dev || true
+# Add CUDA repository and install CUDA toolkit
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    wget \
+    gnupg2 \
+    && wget -qO - https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub | apt-key add - \
+    && echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64 /" > /etc/apt/sources.list.d/cuda.list \
+    && apt-get update \
+    && apt-get install -y --no-install-recommends \
+    cuda-toolkit-12-4 \
+    libcudnn8 \
+    libcublas-12-4 \
+    && rm -rf /var/lib/apt/lists/*
 
-# Install Python 3.10 and required system libraries
-RUN apt-get install -y --no-install-recommends \
-    python3.10 \
-    python3.10-venv \
-    python3.10-dev \
-    python3-pip \
+# Install build tools and system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     gcc \
     g++ \
@@ -20,11 +26,6 @@ RUN apt-get install -y --no-install-recommends \
     elfutils \
     && rm -rf /var/lib/apt/lists/*
 
-# Force Python 3.10 as the default and fix pip
-RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 100 && \
-    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 100 && \
-    update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 100
-
 # Verify Python version
 RUN python --version && pip --version
 
@@ -34,18 +35,19 @@ WORKDIR /code
 COPY requirements.txt .
 
 # Install Python packages with pip upgrade
-RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
-    pip install --no-cache-dir -r requirements.txt
+RUN python3.10 -m pip install --no-cache-dir --upgrade pip setuptools wheel && \
+    python3.10 -m pip install --no-cache-dir -r requirements.txt
 
-# Fix executable stack issue in onnxruntime binary (for Python 3.10 compat)
-RUN find /usr/local/lib/python3.10/site-packages/onnxruntime -name "*.so" -type f -exec \
-    bash -c 'execstack -c "$1" 2>/dev/null || true' _ {} \; || true
+# Fix executable stack issue in onnxruntime binary
+RUN find /usr/local/lib/python3.10/site-packages/onnxruntime -name "*.so" -type f 2>/dev/null | while read f; do \
+    execstack -c "$f" 2>/dev/null || true; \
+done || true
 
 # Fix CUDA library symlinks for onnxruntime compatibility
-RUN find /usr/local/cuda/lib64 -name "libcublas*" -o -name "libcudnn*" && \
-    ln -sf /usr/local/cuda/lib64/libcublasLt.so.12.* /usr/local/cuda/lib64/libcublasLt.so.12 || true && \
-    ln -sf /usr/local/cuda/lib64/libcublas.so.12.* /usr/local/cuda/lib64/libcublas.so.12 || true && \
-    ln -sf /usr/local/cuda/lib64/libcudart.so.12.* /usr/local/cuda/lib64/libcudart.so.12 || true
+RUN find /usr/local/cuda/lib64 -name "libcublas*" -o -name "libcudnn*" 2>/dev/null && \
+    ln -sf /usr/local/cuda/lib64/libcublasLt.so.12.* /usr/local/cuda/lib64/libcublasLt.so.12 2>/dev/null || true && \
+    ln -sf /usr/local/cuda/lib64/libcublas.so.12.* /usr/local/cuda/lib64/libcublas.so.12 2>/dev/null || true && \
+    ln -sf /usr/local/cuda/lib64/libcudart.so.12.* /usr/local/cuda/lib64/libcudart.so.12 2>/dev/null || true
 
 COPY . .