cos-gpu-installer:v2 update nvidia-ctk
We use nvidia-ctk for creating symlinks in /dev/char
for all possible NVIDIA device nodes required for using GPUs in
containers.
This functionality for supporting R550 drivers is first added to
nvidia-cotnainer-toolkit in v1.14.6
BUG=b/326637263
TEST=manual, verify nvidia-ctk working (for gpu driver installation with
R550 drivers).
RELEASE_NOTE=update nvidia-ctk in cos-gpu-installer:v2
Change-Id: I9c7919f6a552970e001635af5d8235f9034d660f
Reviewed-on: https://cos-review.googlesource.com/c/cos/tools/+/67150
Reviewed-by: Shuo Yang <gshuoy@google.com>
Cloud-Build: GCB Service account <228075978874@cloudbuild.gserviceaccount.com>
Tested-by: Arnav Kansal <rnv@google.com>
diff --git a/src/cmd/cos_gpu_installer/Dockerfile b/src/cmd/cos_gpu_installer/Dockerfile
index c1b4655..feb7cbd 100644
--- a/src/cmd/cos_gpu_installer/Dockerfile
+++ b/src/cmd/cos_gpu_installer/Dockerfile
@@ -19,7 +19,7 @@
RUN curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | gpg \
--dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg && \
- curl -s -L https://nvidia.github.io/libnvidia-container/debian11/libnvidia-container.list \
+ curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
| sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
| tee /etc/apt/sources.list.d/nvidia-container-toolkit.list && apt-get update \
&& apt-get install -y nvidia-container-toolkit
