Traceback (most recent call last):
  File ".\train.py", line 13, in <module>
    from utils.callbacks import LossHistory, EvalCallback
  File "C:\Users\cnu\unet-pytorch\utils\callbacks.py", line 17, in <module>
    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 <module>
    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

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐