From 03cfeffca53bd4aa22d4158db449a37132ac851f Mon Sep 17 00:00:00 2001 From: Koushik Dutta Date: Wed, 9 Oct 2024 12:16:22 -0700 Subject: [PATCH] install: nvidia container toolkit --- .../local/install-nvidia-container-toolkit.sh | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 install/local/install-nvidia-container-toolkit.sh diff --git a/install/local/install-nvidia-container-toolkit.sh b/install/local/install-nvidia-container-toolkit.sh new file mode 100644 index 000000000..f1a98b156 --- /dev/null +++ b/install/local/install-nvidia-container-toolkit.sh @@ -0,0 +1,20 @@ +cd ~/.scrypted +docker compose down + +set -e + +# https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html +apt -y update +apt -y install gpg +curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \ + && 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' | \ + sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list +apt -y update +# is there a way to get a versioned package automatically? +apt -y install nvidia-utils-560 +apt -y install nvidia-container-toolkit + +nvidia-ctk runtime configure --runtime=docker +nvidia-ctk config --set nvidia-container-cli.no-cgroups --in-place +systemctl restart docker