Agent skill · NVIDIA
tao-setup-nvidia-gpu-host
Host setup for TAO GPU backends. Checks and, after user approval, installs minimum-compatible NVIDIA driver, CUDA Toolkit, and NVIDIA Container Toolkit versions for Docker/local-Docker and Kubernetes GPU worker hosts. TAO-wide defaults can be overridden by the selected model's runtime profile. The `--check-only` path works on any Linux distribution; `--install` automates debian-family (Ubuntu/Debian/Pop!_OS/Mint/Zorin/Raspbian), rhel-family (Fedora/RHEL/Rocky/AlmaLinux), and suse-family (openSUSE/SLES) hosts, and prints actionable manual-install steps for everything else. Use when the user asks to "set up an NVIDIA GPU host", "check TAO Docker GPU runtime", or prepare a Kubernetes GPU worker for TAO.
What it needs
About 7k tokens when loaded.
What this skill does
NVIDIA GPU Host Setup Standalone install? If this session was not initialized by the TAO skill bank plugin, run the tao-setup skill first (host preflight, credentials, cross-skill discovery). Use this setup skill before TAO workflows run on the docker, local-docker, or kubernetes backend. The TAO-wide default minimums are: NVIDIA driver >=580 (open kernel module preferred) CUDA Toolkit >=13.0 NVIDIA Container Toolkit >=1.19.0 Docker engine — only installed for docker / local-docker backends and only when Docker is missing. The package picked depends on the distro family (docker.io on Debian-family by default, moby-engine / docker-ce from download.docker.com on RHEL-family, docker on SUSE-family). Pass --skip-docker-install to opt out. The check is safe and read-only by default — it works on any Linux distribution because it only probes nvidia-smi, the CUDA toolkit path, the installed container-toolkit package version (via dpkg/rpm/the nvidia-ctk binary version), and the Docker daemon's NVIDIA runtime. Installation must be explicitly authorized by the user and rerun with --install. The install path is automated for these distro families: Family Tested distros Manager Notes --- --- --- --- debian Ubuntu 22.04 / 24.04, Debian 12 (and derivatives Pop!OS, Mint, Zorin, Raspbian, KDE Neon, etc. via UBUNTUCODENAME / VERSIONCODENAME) apt-get Adds NVIDIA cuda-keyring + Container Toolkit .list. Docker via docker.io (override $DOCKERPACKAGEDEBIAN). rhel Fedora 39+, RHEL / Rocky / AlmaLinux 9 and 10 dnf (or yum) Adds NVIDIA cuda-<distro>.repo + Container Toolkit .repo. Docker via Fedora moby-engine when available, otherwise docker-ce from download.docker.com. suse openSUSE Leap 15, SLES 15 zypper Adds the same NVIDIA .repo files. Docker via the distribution docker package. other (Arch, Alpine, Gentoo, NixOS, FreeBSD, …) n/a n/a --install exits with a clear error listing the version targets and the NVIDIA install-guide URLs. Install manually, then rerun --check-only. …
How to use it
Reference it in AdaL, Claude Code, Cursor or any coding agent — nothing to install:
@skills NVIDIA/tao-setup-nvidia-gpu-host