tao-setup-nvidia-gpu-host

NVIDIA GPU Host Setup Use this setup skill before TAO workflows run on the docker , local-docker , or kubernetes backend. It standardizes the host GPU runtime on: NVIDIA driver branch 580 (open kernel module preferred) CUDA Toolkit package 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. Show more Installs 581 Repository nvidia/skills GitHub Stars 1.9K First Seen Jun 8, 2026 Security Audits Gen Agent Trust Hub Pass Socket Pass Snyk Warn