Repository URL to install this package:
|
Version:
1.12.1+cpu ▾
|
import torch
from functools import lru_cache as _lru_cache
def is_built() -> bool:
r"""Returns whether PyTorch is built with MPS support. Note that this
doesn't necessarily mean MPS is available; just that if this PyTorch
binary were run a machine with working MPS drivers and devices, we
would be able to use it."""
return torch._C.has_mps
@_lru_cache()
def is_available() -> bool:
r"""Returns a bool indicating if MPS is currently available."""
return torch._C._is_mps_available()