Rol de Ansible para instalar drivers de NVIDIA

El rol despliega el driver de NVIDIA en máquinas con GPU, que pueden ser baremetal o máquinas virtuales (VM).

Los sistemas operativos soportados son CentOS 7 y Ubuntu 18.04.

Requisitos

La máquina o VM debe tener una GPU de NVIDIA, se puede verificar con:

lspci |grep NVIDIA

Instalación del rol

El rol se puede instalar directamente desde GitHub:

ansible-galaxy install git+https://github.com/LIP-Computing/ansible-role-nvidia.git

Desde Ansible Galaxy:

ansible-galaxy install -f LIP-Computing.ansible-role-nvidia

Variables del rol

Estas variables deben cambiarse dependiendo del tipo de máquina. Específicamente, la versión del driver puede necesitar ser configurada:

• nvidia_driver_version: 440.33.01 - Versión del driver de NVIDIA, verifica el valor predeterminado en defaults/main.yml así como una lista de versiones válidas.
• kmod_install: true - Instalar módulo del kernel, solo para baremetal y VMs.
• dev_create: false - Ejecutar un script para crear dispositivos en el host, solo si kmod_install: true
• nv_persistence: true - Establecer nvidia-smi --persistence-mode=1.

La variable kmod_install habilita la instalación (o no) del módulo del kernel de NVIDIA. En el caso de imágenes de docker, el valor debe ser false, mientras que en el caso de baremetal o VMs debe ser true.

Estas variables tienen valores preconfigurados que no deberían necesitar ser cambiados:

• base_url: URL base de descarga de NVIDIA
• nvidia_driver: Nombre del archivo del driver de NVIDIA, depende de la variable nvidia_driver_version
• nvidia_url: URL completa del driver de NVIDIA, depende de las variables base_url y nvidia_driver

Ejemplo de Playbook

Ejemplo de Playbook: despliegue de la versión del driver 440.33.01 en una VM

- hosts: servers
  roles:
    - { role: LIP-Computing.ansible-role-nvidia, nvidia_driver_version: 40.33.01, kmod_install: true }

Licencia

Apache v2

Información del Autor

• Mario David: mariojmdavid@gmail.com
• LIP Lisboa: http://www.lip.pt
• Proyectos:
  • Indigo DataCloud: https://www.indigo-datacloud.eu/
  • DEEP Hybrid Datacloud: https://deep-hybrid-datacloud.eu/
  • EOSC Synergy: https://www.eosc-synergy.eu/