saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-06-30 03:38:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

mps.is_available -> mps.is_built

Browse source
This commit is contained in:
Quentin Gallouédec 2023-10-06 14:45:24 +02:00
parent 1e5dc90a6a
commit 40ed03cddb
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
stable_baselines3/common/utils.py
View file

@ -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():