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.