9dd3461
1
2
3
4
5
6
7
import torch from importlib.util import find_spec def amp_definitely_not_available(): return not (torch.cuda.is_available() or find_spec('torch_xla'))