GitLab

Do this after the version check to avoid failures on import for earlier
torch where that module did not exist.
See #3901
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>