ai-content-maker/.venv/Lib/site-packages/torch/cuda/amp/common.py

10 lines
202 B
Python

from importlib.util import find_spec
import torch
__all__ = ["amp_definitely_not_available"]
def amp_definitely_not_available():
return not (torch.cuda.is_available() or find_spec("torch_xla"))