From 9d5c4c90d1ac872c8e6692379e00991932a0871e Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Tue, 12 Nov 2024 22:05:41 +0800
Subject: [PATCH] add

---
 ...setup_nvidia_docker_containerd_rootless.sh | 52 +++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100755 nerdctl/setup_nvidia_docker_containerd_rootless.sh

diff --git a/nerdctl/setup_nvidia_docker_containerd_rootless.sh b/nerdctl/setup_nvidia_docker_containerd_rootless.sh
new file mode 100755
index 0000000..9a47e90
--- /dev/null
+++ b/nerdctl/setup_nvidia_docker_containerd_rootless.sh
@@ -0,0 +1,52 @@
+#!/bin/bash
+
+# 获取当前执行脚本的用户
+CURRENT_USER=$(whoami)
+echo "当前用户：$CURRENT_USER"
+
+# 安装 NVIDIA Docker 工具包
+install_nvidia_docker() {
+    echo "正在安装 NVIDIA Docker..."
+    sudo apt-get update
+    sudo apt-get install -y curl gnupg lsb-release
+
+    # 配置 NVIDIA Docker 源
+    if ! grep -q "^deb .\+nvidia-container-toolkit" /etc/apt/sources.list /etc/apt/sources.list.d/*; then
+        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
+    else
+        echo "NVIDIA Docker 源已经配置，跳过此步骤。"
+    fi
+
+    sudo apt-get update
+    sudo apt-get install -y nvidia-container-toolkit
+}
+
+# 配置 NVIDIA Container Toolkit
+configure_nvidia_ctk() {
+    echo "正在配置 NVIDIA Container Toolkit..."
+    
+    # 创建配置目录
+    mkdir -p "$HOME/.config/containerd"
+
+    # 配置 nvidia-ctk 到 rootless 模式下的 config.toml
+    nvidia-ctk runtime configure --runtime=containerd --config="$HOME/.config/containerd/config.toml"
+}
+
+# 重启 containerd 服务
+restart_containerd() {
+    echo "重启 containerd 服务..."
+    systemctl --user daemon-reload
+    systemctl --user restart containerd
+    sudo systemctl daemon-reload
+    sudo systemctl restart containerd
+}
+
+# 执行所有步骤
+install_nvidia_docker
+configure_nvidia_ctk
+restart_containerd
+
+echo "所有步骤已完成，NVIDIA Docker 和 containerd 配置已更新。"