From 43b459e4ba482bf4916ce3b5628c2a725e31dffc Mon Sep 17 00:00:00 2001 From: Akiva Berger Date: Thu, 19 Dec 2024 10:20:29 +0200 Subject: [PATCH] chore(Docker): upgrade linker to cuda 12 --- build/linker/Dockerfile | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/build/linker/Dockerfile b/build/linker/Dockerfile index 4be52f7b59..af26905c7d 100644 --- 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/*