Use ResourceInformation::hasGpuNvidia() in TensorFlow

Author: makortel

PhysicsTools/TensorFlow/src/TensorFlow.cc

@@ -41,7 +41,7 @@ namespace tensorflow {
     }
     // NVidia GPU
     else if (backend == Backend::cuda) {
-      if (not ri->nvidiaDriverVersion().empty()) {
+      if (ri->hasGpuNvidia()) {
         // Check if one GPU device is visible to TF
         // If not, an exception is raised --> this can happen in case of driver version mismatch
         // or missing CUDA support in TF compilation
@@ -73,7 +73,7 @@ namespace tensorflow {
     // Get NVidia GPU if possible or fallback to CPU
     else if (backend == Backend::best) {
       // Check if a Nvidia GPU is availabl
-      if (not ri->nvidiaDriverVersion().empty()) {
+      if (ri->hasGpuNvidia()) {
         // Take only the first GPU in the CUDA_VISIBLE_DEVICE list
         (*_options.config.mutable_device_count())["GPU"] = 1;
         _options.config.mutable_gpu_options()->set_visible_device_list("0");