From b3c948402b582466c6e10834973cf31348780c9a Mon Sep 17 00:00:00 2001 From: shenlan Date: Fri, 27 Jun 2025 01:04:54 +0800 Subject: [PATCH] chore: add nvidia repo setup to offline package script --- scripts/create-gpu-k8s-offline-package.sh | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/scripts/create-gpu-k8s-offline-package.sh b/scripts/create-gpu-k8s-offline-package.sh index 5994f8b..5ad745e 100755 --- a/scripts/create-gpu-k8s-offline-package.sh +++ b/scripts/create-gpu-k8s-offline-package.sh @@ -28,6 +28,14 @@ APT_PACKAGES=( containerd "$NVIDIA_DRIVER_VERSION" nvidia-container-toolkit ) +# Add NVIDIA repository for nvidia-container-toolkit +distribution=$(. /etc/os-release; echo $ID$VERSION_ID) +curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \ + sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg +curl -s -L "https://nvidia.github.io/nvidia-docker/${distribution}/nvidia-docker.list" | \ + sed 's#^deb #deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] #' | \ + sudo tee /etc/apt/sources.list.d/nvidia-docker.list + sudo apt-get update -y sudo apt-get install --download-only -y "${APT_PACKAGES[@]}" cp /var/cache/apt/archives/*.deb "$WORKDIR/packages/"