Replaced deprecated is_gpu_available with `list_physical_devices("G…

…PU")`
bethgelab · Dec 26, 2024 · d2fbda7 · d2fbda7
1 parent 339aa06
commit d2fbda7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -266,7 +266,7 @@ def tensorflow_resnet50(request: Any) -> ModelAndData:
 
     import tensorflow as tf
 
-    if not tf.test.is_gpu_available():
+    if not list_physical_devices("GPU"):
         pytest.skip("ResNet50 test too slow without GPU")
 
     model = tf.keras.applications.ResNet50(weights="imagenet")