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