目的:使用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

 

这样就可以了!

 

 

 

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐