8 lines
136 B
Python
8 lines
136 B
Python
|
import torch
|
||
|
|
|
||
|
|
__all__ = ["amp_definitely_not_available"]
|
||
|
|
|
||
|
|
|
||
|
|
def amp_definitely_not_available():
|
||
|
|
return not torch.vacc.is_available()
|