Update Dockerfile

Dockerfile

@@ -16,7 +16,7 @@ RUN apt-get update && \
 RUN wget https://developer.download.nvidia.com/compute/cuda/repos/debian12/x86_64/cuda-keyring_1.1-1_all.deb && \
     dpkg -i cuda-keyring_1.1-1_all.deb && \
     apt-get update && \
-    apt-get -y install cuda-toolkit-12-2
+    apt-get -y install cuda
 
 # CUDA ENV-Variablen setzen
 ENV PATH=/usr/local/cuda/bin:${PATH}