update

Author: OutisLi

deepmd/tf/train/run_options.py

@@ -81,11 +81,15 @@ def get_device_name(self) -> str | None:
         str or None
             The device name if available, otherwise None.
         """
-        gpus = tf.config.list_physical_devices("GPU")
-        if gpus:
-            # Use the first physical GPU device identifier as the device name
-            details = tf.config.experimental.get_device_details(gpus[0])
-            return details.get("device_name") or gpus[0].name
+        try:
+            gpus = tf.config.get_visible_devices("GPU")
+            if gpus:
+                # Use the first visible GPU device identifier as the device name
+                details = tf.config.experimental.get_device_details(gpus[0])
+                return details.get("device_name") or gpus[0].name
+        except (AttributeError, RuntimeError):
+            # Experimental API may not exist or fail in some TF versions
+            pass
         return None