Skip to content

Only enable GPU runtime if available

Samuel GAIST requested to merge only_enable_gpu_runtime_if_available into master

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