Update Dockerfile

Dockerfile

@@ -1,23 +1,21 @@
-FROM nvidia/cuda:12.1.1-base-ubuntu22.04
+# Use a lightweight Python base image
+FROM python:3.10-slim
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
-    python3.10 \
-    python3-pip \
     git \
     curl \
     cmake \
     build-essential \
-    libssl-dev \
     libopenblas-dev \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# Upgrade pip and install setuptools first
+# Upgrade pip and install setuptools
 RUN pip install --no-cache-dir --upgrade pip setuptools
 
-# Install requirements
+# Install Python requirements
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt