Source code for super_gradients.training.utils.version_utils
import torch
_TORCH_VERSION_MAJOR_MINOR_ = tuple(map(int, torch.version.__version__.split(".")[:2]))
__all__ = ["torch_version_is_greater_or_equal"]
[docs]def torch_version_is_greater_or_equal(major: int, minor: int) -> bool:
version = (major, minor)
return _TORCH_VERSION_MAJOR_MINOR_ >= version