chore(Docker): upgrade linker to cuda 12

Sefaria · Dec 19, 2024 · 43b459e · 43b459e
1 parent 0f14d8a
commit 43b459e
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/build/linker/Dockerfile b/build/linker/Dockerfile
@@ -25,15 +25,12 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/
 
 RUN apt-get update -qq \
  && apt-get install --no-install-recommends -y \
-    # install cuda toolkit
-    cuda-libraries-11-6 \
-    libnpp-11-6 \
-    cuda-nvtx-11-6 \
-    libcublas-11-6 \
+    cuda-libraries-12-2 \
+    libnpp-12-2 \
+    cuda-nvtx-12-2 \
+    libcublas-12-2 \
     libnccl2 \
-    # install cudnn
-    libcudnn8=8.4.0.27-1+cuda11.6 \
- && apt-mark hold libcublas-11-6 \
+    libcudnn8 \
  && apt-get clean \
  && rm -rf /var/lib/apt/lists/*