mps.is_available -> mps.is_built

DLR-RM · Oct 6, 2023 · 40ed03c · 40ed03c
1 parent 1e5dc90
commit 40ed03c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stable_baselines3/common/utils.py b/stable_baselines3/common/utils.py
@@ -521,7 +521,7 @@ def get_available_accelerator() -> str:
     Return the available accelerator
     (currently checking only for CUDA and MPS device)
     """
-    if hasattr(th, "has_mps") and th.backends.mps.is_available():
+    if hasattr(th, "has_mps") and th.backends.mps.is_built():
         # MacOS Metal GPU
         return "mps"
     elif th.cuda.is_available():