utils.py 347 B

1234567891011121314
  1. # Copyright (c) Microsoft Corporation.
  2. # SPDX-License-Identifier: Apache-2.0
  3. # DeepSpeed Team
  4. from deepspeed.utils.torch import required_torch_version
  5. def is_torch_elastic_compatible():
  6. '''
  7. Helper to lookup torch version. Elastic training is
  8. introduced in 1.11.x
  9. '''
  10. return required_torch_version(min_version=1.11)