Explicitly add CUDA

Dockerfile

@@ -1,18 +1,9 @@
 FROM nvidia/cuda:12.4.1-cudnn8-runtime-ubuntu22.04
 
-# Install Python and ensure pip is up to date
+# Install Python and required system libraries
 RUN apt-get update && apt-get install -y --no-install-recommends \
     python3.10 \
     python3-pip \
-    && rm -rf /var/lib/apt/lists/*
-
-# Create Python 3.10 symlink
-RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1
-
-WORKDIR /code
-
-# Install build tools and system dependencies
-RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     gcc \
     g++ \
@@ -21,12 +12,19 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libsm6 \
     libxext6 \
     libxrender-dev \
+    cuda-runtime-12-4 \
+    libcublas-12-4 \
     && rm -rf /var/lib/apt/lists/*
 
+# Create Python 3.10 symlink
+RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1
+
+WORKDIR /code
+
 # Copy requirements first for better layer caching
 COPY requirements.txt .
 
-# Install Python packages
+# Install Python packages with pip upgrade
 RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
     pip install --no-cache-dir -r requirements.txt
 
@@ -35,12 +33,10 @@ COPY . .
 # Set environment variables for CUDA
 ENV CUDA_HOME=/usr/local/cuda
 ENV PATH=${CUDA_HOME}/bin:${PATH}
-ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${CUDA_HOME}/extras/CUPTI/lib64:${LD_LIBRARY_PATH}
+ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:/usr/local/cuda-12.4/lib64:${LD_LIBRARY_PATH}
 ENV INSIGHTFACE_HOME=/tmp/.insightface
 ENV MPLCONFIGDIR=/tmp/matplotlib
 
 EXPOSE 7860
 
-CMD ["python", "-m", "uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]EXPOSE 7860
-
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["python", "-m", "uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]