TensorBoard使用报错
TensorBoard使用报错
1.报错如下
Traceback (most recent call last):
File "E:\AI_DP\xtd\3-Tensorboard用途.py", line 5, in <module>
from torch.utils.tensorboard import SummaryWriter
File "E:\AI\anaconda\envs\DP_learn\lib\site-packages\torch\utils\tensorboard\__init__.py", line 4, in <module>
LooseVersion = distutils.version.LooseVersion
AttributeError: module 'distutils' has no attribute 'version'
进程已结束,退出代码1
2.解决方法
安装 TensorBoard 2.4 版本:在激活的环境中,运行以下命令来安装指定版本的 TensorBoard:
conda install -c anaconda tensorboard=2.4.0
最后更改代码
修改 PyTorch 的 TensorBoard 接口文件(推荐)
-
定位问题文件:错误显示发生在
E:\AI\anaconda\envs\DP_learn\lib\site-packages\torch\utils\tensorboard\__init__.py
的第 4 行 -
修改文件内容:
-
打开上述文件
-
将第 4 行的
LooseVersion = distutils.version.LooseVersion
修改为
from packaging.version import Version as LooseVersion
-
-
确保安装了
packaging
库:pip install packaging