当使用PyTorch 1.4和TensorBoard时，发现TensorBoard无法显示计算图，只显示scalar。经确认，此问题由PyTorch版本过高引起，支持graph的最高版本为1.3.1。解决方案包括：1) 卸载并安装PyTorch 1.3.1；2) 对应地降级torchvision到0.4版本；3) 解决因PILLOW_VERSION导入错误导致的问题，该问题在Pillow 7.0.0之后的版本中被移除，需调整相关调用。按照以上步骤操作后，计算图成功显示。
电脑安装的pytoch版本为1.4，tensorboard无法显示计算图graph，但可以显示scalar。
经过多方查证为pytorch版本问题，目前graph显示仅支持到pytorch1.3。
处理如下：
（1）卸载当前的pytorch并安装低版本1.3.1版本
卸载当前的高版本，cmd中输入：
pip uninstall torch
安装低版本，采用离线安装的方式。
从连接https://download.pytorch.org/whl/torch_stable.html
下载1.3.1版本pytorch torch-1.3.1+cpu-cp37-cp37m-win_amd64.whl
离线安装，cmd中输入：
D:\download\torch-1.3.1+cpu-cp37-cp37m-win_amd64.whl
最低0.47元/天 解锁文章
扫一扫
1833
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
