livebook-dev/livebook
1
1
Fork 0
mirror of https://github.com/livebook-dev/livebook.git synced 2025-09-04 12:04:20 +08:00
Code Issues Projects Releases Packages Wiki Activity

Bump Docker images to CUDA 12.8 and include NCCL (#3018)

Browse source
This commit is contained in:
Jonatan Kłosko 2025-06-17 21:01:09 +02:00 • committed by GitHub
parent 231b30d464
commit 31ff347768
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
.github/workflows/release.yml vendored
View file

@ -75,7 +75,8 @@ jobs:
tag_suffix: "-cuda12"
build_args: |
VARIANT=cuda
CUDA_VERSION=12-6
CUDA_VERSION_MAJOR=12
CUDA_VERSION_MINOR=8
steps:
- uses: actions/checkout@v4
- run: |

6
Dockerfile
View file

@ -7,7 +7,8 @@ FROM ${BASE_IMAGE} AS base-default
FROM ${BASE_IMAGE} AS base-cuda
ARG CUDA_VERSION
ARG CUDA_VERSION_MAJOR
ARG CUDA_VERSION_MINOR
RUN distro="ubuntu$(. /etc/lsb-release; echo "$DISTRIB_RELEASE" | tr -d '.')" && \
# Official Docker images use the sbsa packages when targetting arm64.
@ -20,8 +21,7 @@ RUN distro="ubuntu$(. /etc/lsb-release; echo "$DISTRIB_RELEASE" | tr -d '.')" &&
# the CUDA toolkit that is required by Elixir numerical packages
# (nvcc and runtime libraries). Note that we do not need to install
# the driver, it is already provided by NVIDIA Container Toolkit.
cuda_version="${CUDA_VERSION}" && cuda_major="${cuda_version%-*}" && \
apt-get install -y git cuda-nvcc-${CUDA_VERSION} cuda-libraries-${CUDA_VERSION} libcudnn9-cuda-$cuda_major && \
apt-get install -y git cuda-nvcc-${CUDA_VERSION_MAJOR}-${CUDA_VERSION_MINOR} cuda-libraries-${CUDA_VERSION_MAJOR}-${CUDA_VERSION_MINOR} libcudnn9-cuda-${CUDA_VERSION_MAJOR} libnccl2=*+cuda${CUDA_VERSION_MAJOR}.${CUDA_VERSION_MINOR} && \
apt-get clean -y && rm -rf /var/lib/apt/lists/*
ENV PATH="/usr/local/nvidia/bin:/usr/local/cuda/bin:$PATH"