install lib cuda 12

kermitt2 · Sep 24, 2024 · db41747 · db41747
1 parent 980a7ac
commit db41747
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.delft b/Dockerfile.delft
@@ -72,7 +72,7 @@ FROM tensorflow/tensorflow:2.7.0-gpu
 ENV LANG C.UTF-8
 
 # update NVIDIA Cuda key (following a key rotation in April 2022)
-RUN apt-get install -y wget && \
+RUN apt-get install -y libcublas-12-0 && \
     apt-key del 7fa2af80 && \
     rm /etc/apt/sources.list.d/cuda.list && \
     rm /etc/apt/sources.list.d/nvidia-ml.list && \