Add build-essential and upgrade pip to ensure packages install correctly

Dockerfile

@@ -1,18 +1,19 @@
 FROM nvidia/cuda:12.1.0-base-ubuntu22.04
 
-# Install Python 3.11 and pip
+# Install Python 3.11, pip, and build tools
 RUN apt-get update && \
-    apt-get install -y python3.11 python3.11-venv python3-pip && \
+    apt-get install -y python3.11 python3.11-dev python3-pip build-essential && \
     ln -sf /usr/bin/python3.11 /usr/bin/python && \
     ln -sf /usr/bin/pip3 /usr/bin/pip && \
+    pip install --upgrade pip setuptools wheel && \
     rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
 # Install dependencies
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt \
- && rm -rf ~/.cache/pip
+RUN pip install --no-cache-dir -r requirements.txt && \
+    rm -rf ~/.cache/pip
 
 # Copy source code
 COPY . .