GPU Install Role

This Role installs GPU drivers for AMD and Nvidia Graphics cards.

Requirements

None

Role Variables

• gpu_install_vendor
  • The vendor of the GPU
  • Valid values:
    • amd
    • nvidia
• gpu_install_library
  • Whether or not to install supporting GPU Libraries
    • CUDA for Nvidia
    • ROCm for AMD
  • Valid values
    • true to install supporting libraries
    • false to skip installation of supporting libraries (Default)
• gpu_install_driver_version
  • The version of the relevant drivers to install
  • Valid values depend on vendor
    • Default for Nvidia is 550.127.05
    • Default for AMD is 6.3.3
• gpu_install_library_version
  • Installs CUDA/ROCm with the specified version
  • Valid values depend on vendor
    • Default for Nvidia is 12.4
    • Default for AMD is 6.3.3
• gpu_install_dry_run
  • Perform a dry run without actually installing anything
  • Default is False

Dependencies

None

Example Playbook

Nvidia

---
- name: Install Drivers
  hosts: all
  become: True
  roles:
    - role: gpu_install
      gpu_install_vendor: nvidia
      gpu_install_driver_version: 550.127.05
      gpu_install_library_version: 12.4

AMD

---
- name: Install Drivers
  hosts: all
  become: True
  roles:
    - role: gpu_install
      gpu_install_vendor: amd
      gpu_install_library_version: 6.3.3

License

GPL-2.0

Author Information

An optional section for the role authors to include contact information, or a website (HTML is not allowed).