We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d47c586 commit b85a2a5Copy full SHA for b85a2a5
stable_baselines3/common/utils.py
@@ -521,7 +521,7 @@ def get_available_accelerator() -> str:
521
Return the available accelerator
522
(currently checking only for CUDA and MPS device)
523
"""
524
- if hasattr(th, "has_mps") and th.backends.mps.is_built():
+ if hasattr(th, "backends") and th.backends.mps.is_built():
525
# MacOS Metal GPU
526
return "mps"
527
elif th.cuda.is_available():
0 commit comments