File size: 157 Bytes
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'))