AttributeError: module ‘distutils‘ has no attribute ‘version‘解决跑pytorch代码报错

Traceback (most recent call last):
  File ".\train.py", line 13, in 
    from utils.callbacks import LossHistory, EvalCallback
  File "C:\Users\cnu\unet-pytorch\utils\callbacks.py", line 17, in 
    from torch.utils.tensorboard import SummaryWriter
  File "D:\Program Files\Anaconda\envs\unet_pytorch\lib\site-packages\torch\utils\tensorboard\__init__.py", line 4, in 
    LooseVersion = distutils.version.LooseVersion
AttributeError: module 'distutils' has no attribute 'version'

运行代码遇到上面的问题:

解决方法:网上建议 :pip install setuptools==59.5.0

这里建议修改代码将代码注释

import tensorboard
from setuptools import distutils

# LooseVersion = distutils.version.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

from .writer import FileWriter, SummaryWriter  # noqa: F401
from tensorboard.summary.writer.record_writer import RecordWriter  # noqa: F401

你可能感兴趣的:(pytorch,数学建模,人工智能)