26 lines
846 B
YAML
26 lines
846 B
YAML
---
|
|
# Fail if no NVIDIA hardware is detected.
|
|
nvidia_runtime_require_gpu: true
|
|
|
|
# Install/repair NVIDIA driver packages.
|
|
nvidia_runtime_install_driver: true
|
|
|
|
# Optional explicit driver package pin (for example: nvidia-driver-550).
|
|
# When empty on Ubuntu, the role will auto-detect the recommended package.
|
|
nvidia_runtime_driver_package: ""
|
|
|
|
# Install CUDA toolkit from distro repository.
|
|
nvidia_runtime_install_cuda_toolkit: false
|
|
nvidia_runtime_cuda_package: nvidia-cuda-toolkit
|
|
|
|
# Install NVIDIA container runtime package if available in configured repos.
|
|
nvidia_runtime_install_container_toolkit: false
|
|
nvidia_runtime_container_toolkit_package: nvidia-container-toolkit
|
|
|
|
# Reboot handling
|
|
nvidia_runtime_reboot_if_needed: false
|
|
nvidia_runtime_reboot_timeout: 900
|
|
|
|
# Post-install validation behavior
|
|
nvidia_runtime_validate_after_install: true
|