lightning/requirements/check-avail-strategies.py

8 lines
169 B
Python

import bagua # noqa: F401
import deepspeed # noqa: F401
import fairscale # noqa: F401
import horovod.torch
# returns an error code
assert horovod.torch.nccl_built()