You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
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"))
|
