Describe the bug
the current utilities for identifying pytorch versions are not robust to those from nvidia pytorch docker images (such as PT_BEFORE_1_7, version_leq(torch.__version__, '1.10')). Given that these are important for compatibility with different pytorch versions, the utility should be unified and tested.