use base_version to check torch version in torch_less_than_1_11 (#16806)

* use base_version

* make is_torch_less_than_1_8 match 1_11

Co-authored-by: Nicholas Broad <[email protected]>

Author: nbroad1881

src/transformers/pytorch_utils.py

@@ -23,8 +23,8 @@
 
 logger = logging.get_logger(__name__)
 
-is_torch_less_than_1_8 = version.parse(torch.__version__) < version.parse("1.8.0")
-is_torch_less_than_1_11 = version.parse(torch.__version__) < version.parse("1.11")
+is_torch_less_than_1_8 = version.parse(version.parse(torch.__version__).base_version) < version.parse("1.8.0")
+is_torch_less_than_1_11 = version.parse(version.parse(torch.__version__).base_version) < version.parse("1.11")
 
 
 def torch_int_div(tensor1, tensor2):