/
usr
/
local
/
lib64
/
python3.6
/
site-packages
/
torch
/
cuda
/
amp
/
/usr/local/lib64/python3.6/site-packages/torch/cuda/amp
mkdir
upload
Name
Size
Mode
Actions
__pycache__/
-
0755
rm
autocast_mode.py
4562
0644
edit
dl
rm
common.py
157
0644
edit
dl
rm
grad_scaler.py
25315
0644
edit
dl
rm
__init__.py
124
0644
edit
dl
rm
Edit:
/usr/local/lib64/python3.6/site-packages/torch/cuda/amp/common.py
(157B)
import torch from importlib.util import find_spec def amp_definitely_not_available(): return not (torch.cuda.is_available() or find_spec('torch_xla'))
Save
import torch from importlib.util import find_spec def amp_definitely_not_available(): return not (torch.cuda.is_available() or find_spec('torch_xla'))