ai-workspace-infra/gitops
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix NVIDIA runtime install

Browse Source
This commit is contained in:
shenlan 2025-06-25 20:28:19 +08:00
parent bb81e13de9
commit 85caf2f3b0
3 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/gpu-k8s-role.md
View File

@ -7,7 +7,7 @@ This document describes how to use the `gpu-k8s` role to deploy a simple Kuberne
The role performs three main tasks: The role performs three main tasks:
1. **Create the Kubernetes cluster** using [sealos](https://github.com/labring/sealos). It runs the provided `sealos run` command to bootstrap the master and worker nodes. 1. **Create the Kubernetes cluster** using [sealos](https://github.com/labring/sealos). It runs the provided `sealos run` command to bootstrap the master and worker nodes.
2. **Install NVIDIA drivers and container runtime** on the target hosts so that Kubernetes can access GPU resources. 2. **Install NVIDIA drivers and the NVIDIA container toolkit** on the target hosts so that Kubernetes can access GPU resources.
3. **Verify GPU access** by deploying the official NVIDIA device plugin and running a small CUDA workload. 3. **Verify GPU access** by deploying the official NVIDIA device plugin and running a small CUDA workload.

10
playbooks/roles/vhosts/common/tasks/include_gpu.yaml
View File

@ -1,9 +1,11 @@
- name: Add NVIDIA repository - name: Add NVIDIA repositories
shell: | shell: |
add-apt-repository -y ppa:graphics-drivers add-apt-repository -y ppa:graphics-drivers
curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | tee /etc/apt/sources.list.d/nvidia-container-runtime.list curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | apt-key add -
curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | tee /etc/apt/sources.list.d/nvidia-container-runtime.list
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list
apt-get update apt-get update
- name: Install NVIDIA driver and container runtime - name: Install NVIDIA driver and container runtime
@ -12,6 +14,6 @@
- nvidia-modprobe - nvidia-modprobe
- nvidia-driver-535 - nvidia-driver-535
- nvidia-headless-535 - nvidia-headless-535
- nvidia-container-runtime - nvidia-container-toolkit
state: present state: present
update_cache: yes update_cache: yes

10
playbooks/roles/vhosts/gpu-k8s/tasks/install_driver.yml
View File

@ -1,9 +1,11 @@
- name: Add NVIDIA repository - name: Add NVIDIA repositories
shell: | shell: |
add-apt-repository -y ppa:graphics-drivers add-apt-repository -y ppa:graphics-drivers
curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | tee /etc/apt/sources.list.d/nvidia-container-runtime.list curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | apt-key add -
curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | tee /etc/apt/sources.list.d/nvidia-container-runtime.list
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list
apt-get update apt-get update
args: args:
executable: /bin/bash executable: /bin/bash
@ -15,7 +17,7 @@
- nvidia-modprobe - nvidia-modprobe
- nvidia-driver-535 - nvidia-driver-535
- nvidia-headless-535 - nvidia-headless-535
- nvidia-container-runtime - nvidia-container-toolkit
state: present state: present
update_cache: yes update_cache: yes
become: true become: true