diff --git a/docker/Dockerfile b/docker/Dockerfile
index ded61b5..85fbf0a 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,26 +1,39 @@
 # Use an official CUDA runtime with Ubuntu as a parent image
 FROM nvidia/cuda:12.2.0-devel-ubuntu22.04
 
+# Set the working directory in the container
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    curl \
+    ca-certificates \
+    && rm -rf /var/lib/apt/lists/*
+
 # Install Python and pip
-RUN apt-get update && apt-get install -y \
+RUN apt-get update && apt-get install -y --no-install-recommends \
     python3.11 \
     python3-pip \
     && rm -rf /var/lib/apt/lists/*
 
-# Set the working directory in the container
-WORKDIR /usr/src/app
+# Upgrade pip
+RUN pip3 install --no-cache-dir --upgrade pip
 
 # Get requirements
-COPY requirements.txt requirements.txt
+COPY pyproject.toml .
 
-# Install packages specified in requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
+# Install packages specified in pyproject.toml cu121
+RUN pip3 install --no-cache-dir .[cu121]
 
-# Copy the current directory contents into the container at /usr/src/app
+# Copy the current directory contents into the container
 COPY . .
 
 # Make port 5000 available to the world outside this container
 EXPOSE 5000
 
+# Set the entry point
+ENTRYPOINT ["python3"]
+
 # Run main.py when the container launches
-CMD ["python3", "main.py"]
+CMD ["main.py"]
\ No newline at end of file