Disable the use of GPU from tensorflow by default

cms-sw · Jan 17, 2023 · 2fa5675 · 2fa5675
1 parent c628e55
commit 2fa5675
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/PhysicsTools/TensorFlow/src/TensorFlow.cc b/PhysicsTools/TensorFlow/src/TensorFlow.cc
@@ -85,6 +85,10 @@ namespace tensorflow {
     // objects to create the session
     Status status;
 
+    // hotfix: disable GPU usage whatsoever for now, add a convenient interface in a future PR
+    (*sessionOptions.config.mutable_device_count())["GPU"] = 0;
+    sessionOptions.config.mutable_gpu_options()->set_visible_device_list("");
+
     // create a new, empty session
     Session* session = nullptr;
     status = NewSession(sessionOptions, &session);