Fix distutils.LooseVersion DeprecationWarning (#88524)

Jasha10 · pytorchmergebot · commit 1c681f4bd8fb · 2022-12-27T11:46:00.000Z
Fixes #84712 Pull Request resolved: #88524 Approved by: https://github.com/MaKaNu, https://github.com/milutter, https://github.com/soumith
diff --git a/setup.py b/setup.py
@@ -981,6 +981,7 @@ def main():
         'typing_extensions',
         'sympy',
         'networkx',
+        'packaging'
     ]
 
     extras_require = {
diff --git a/torch/utils/tensorboard/__init__.py b/torch/utils/tensorboard/__init__.py
@@ -1,12 +1,12 @@
 import tensorboard
-from distutils.version import LooseVersion
+from packaging import version  # type: ignore[import]
 
-if not hasattr(tensorboard, "__version__") or LooseVersion(
+if not hasattr(tensorboard, "__version__") or version.parse(
     tensorboard.__version__
-) < LooseVersion("1.15"):
+) < version.Version("1.15"):
     raise ImportError("TensorBoard logging requires TensorBoard version 1.15 or above")
 
-del LooseVersion
+del version
 del tensorboard
 
 from .writer import FileWriter, SummaryWriter  # noqa: F401
