torch_version_ge

torchtune.utils.torch_version_ge(version: str) → bool

检查 torch 版本是否大于或等于给定版本。

参数:

version (str) – 要与之比较的 torch 版本

返回:

如果 torch 版本大于或等于给定版本，则为 True。

返回类型:

bool

示例

>>> print(torch.__version__)
2.4.0
>>> torch_version_ge("2.0")
True