Skip to content

Only enable GPU runtime if available

Some execution environments may run CPU as well as GPU code, this check for Nvidia driver presence will allow to reuse the same environment on workers with and without GPU available.

It also updates the GPU option to use the latest version provided by the nvidia-docker2 documentation.

Merge request reports

Loading