fix

Dockerfile

@@ -13,6 +13,21 @@ ENV PYTHONPATH="/srv"
 RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
 
 RUN apt-get -y update
+RUN apt-get install -y --no-install-recommends \
+    build-essential \
+    python3-dev \
+    libglib2.0-0 \
+    libgl1-mesa-glx \
+    libpq-dev \
+    make \
+    cuda
+
+# Add NVIDIA CUDA repository
+RUN apt-get update && apt-get install -y wget
+RUN wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.1-1_all.deb
+RUN dpkg -i cuda-keyring_1.1-1_all.deb
+RUN apt-get update
+
 RUN apt-get install -y --no-install-recommends \
     build-essential \
     python3-dev \