GPU Support

OSD supports the installation of GPU drivers for the compute nodes via a set of Juju Actions.

We currently only support Nvidia GPUs.

Nvidia

The slurmd charm provides an nvidia-install action to install the Nvidia drivers easily:

$ juju run-action slurmd/leader nvidia-install

This action, by default, pulls the latest drivers from official Nvidia repositories. If you need to install older drivers, you should specify them with the action nvidia-package before installing the drivers. For example, to install the version 470 for the Kepler GPUs:

$ juju run-action slurmd/leader nvidia-package package="nvidia-driver-branch-470" # for CentOS7
$ juju run-action slurmd/leader nvidia-package package="cuda-drivers-470" # for Ubuntu 20.04
$ juju run-action slurmd/leader nvidia-install

Please check the official Nvidia documentation about compatibility before installing the drivers.

Note

By default, OSD will install the latest drivers from Nvidia. This might not be the drivers your graphics card need. Please double check the versions before installing the drivers. Nvidia website has a compatibility page to guide you on installing the correct version.

After installing GPU Drivers, your need to reboot the node. Check Rebooting Nodes section for details on rebooting them.