common.py 200 B

1234567
  1. import torch
  2. from importlib.util import find_spec
  3. __all__ = ["amp_definitely_not_available"]
  4. def amp_definitely_not_available():
  5. return not (torch.cuda.is_available() or find_spec('torch_xla'))