Skip to content

Commit

Permalink
Merge pull request #2194 from Sefaria/up_cuda_12
Browse files Browse the repository at this point in the history
chore(Docker): upgrade linker to cuda 12
  • Loading branch information
akiva10b authored Dec 19, 2024
2 parents 0f14d8a + 43b459e commit 16a8ada
Showing 1 changed file with 5 additions and 8 deletions.
13 changes: 5 additions & 8 deletions build/linker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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/*

Expand Down

0 comments on commit 16a8ada

Please sign in to comment.