Source code for torch.backends.mps

import torch
from functools import lru_cache as _lru_cache

[docs]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
[docs]@_lru_cache() def is_available() -> bool: r"""Returns a bool indicating if MPS is currently available.""" return torch._C._is_mps_available()


Access comprehensive developer documentation for PyTorch

View Docs


Get in-depth tutorials for beginners and advanced developers

View Tutorials


Find development resources and get your questions answered

View Resources