wehub-resource-sync
15 lines
347 B
Python
15 lines
347 B
Python
# Copyright (c) Microsoft Corporation.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
# DeepSpeed Team
|
|
|
|
from deepspeed.utils.torch import required_torch_version
|
|
|
|
|
|
def is_torch_elastic_compatible():
|
|
'''
|
|
Helper to lookup torch version. Elastic training is
|
|
introduced in 1.11.x
|
|
'''
|
|
return required_torch_version(min_version=1.11)
|