diff --git a/requirements.txt b/requirements.txt index c9f16ce575632..f2003b0a3780f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,6 +3,7 @@ astunparse expecttest future numpy +packaging psutil pyyaml requests diff --git a/torch/utils/tensorboard/__init__.py b/torch/utils/tensorboard/__init__.py index bb1885d756c50..6c668d894cae1 100644 --- a/torch/utils/tensorboard/__init__.py +++ b/torch/utils/tensorboard/__init__.py @@ -1,12 +1,9 @@ import tensorboard -from setuptools import distutils - -LooseVersion = distutils.version.LooseVersion +from packaging.version import Version as LooseVersion if not hasattr(tensorboard, '__version__') or LooseVersion(tensorboard.__version__) < LooseVersion('1.15'): raise ImportError('TensorBoard logging requires TensorBoard version 1.15 or above') -del distutils del LooseVersion del tensorboard