Pytorch | No module named 'torch.utils.tensorboard' 找不到tensorboard

 

目的：使用tensorboard可视化Pytorch模型

进展：Pytorch1.1.0版本，并使用pip install tensorboard

报错信息：No module named 'torch.utils.tensorboard' 提示找不到tensorboard

 

搜到的原解决方案：https://www.endtoend.ai/blog/pytorch-tensorboard/

问题分析：主要由于tensorboard和pytorch版本不匹配，首先升级pytorch，然后再升级安装tensorboard。

具体操作：

pip install --upgrade torch

输入以上指令升级pytorch，然后

pip install --upgrade tensorboard

顺带安装future

pip install future

 

这样就可以了！