pytorch1.4+tensorboard不显示graph计算图的问题

当使用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之后的版本中被移除,需调整相关调用。按照以上步骤操作后,计算图成功显示。
摘要由CSDN通过智能技术生成

电脑安装的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

(2)安装完成pytorch后发现其匹配的torchvision为0.4版本,参考(1)对torchvision进行降版本<

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值