using_torch_2

composer.utils.using_torch_2()

Check the PyTorch version and compared it with version 2.0.0.

Returns

bool – Return True if current version is greater than or equal to 2.0.0 else False