pytorch笔记--tensorboardX出错:不能显示graph网络结构
程序员文章站
2022-03-26 23:12:27
本机环境 Win10,1660Ti,VS2017,Python3.7.6,CUDA10.0.130,cuDNN7.6.5,tensorflow1.15,tensorboard1.15,tensorboardX 2.1解决办法 如果出现与该博客相似的问题,即无法显示网络结构graph,应该就是torch版本问题,在此对该博客进行补充,由于我的 cuda 版本10.0,所以在选择torch版本下载的时候,只有 torch1.2(cu100) 和 torch1.1(cu100)合适,我尝试过 torch...
本机环境
Win10,1660Ti,VS2017,Python3.7.6,CUDA10.0.130,cuDNN7.6.5,tensorflow1.15-cpu,tensorboard1.15,tensorboardX 2.1
解决办法
如果出现与该博客相似的问题,即无法显示网络结构graph,应该就是torch版本问题,在此对该博客进行补充,由于我的 cuda 版本10.0,所以在选择torch版本下载的时候,只有 torch1.2(cu100) 和 torch1.1(cu100)合适,我尝试过 torch1.2(cu100) 同样存在无法显示graph的问题。最终的版本是:torch1.1(cu100) 和 torchvision4.0(cu100),所以下调 torch 版本需要考虑 cuda 版本,如果直接使用 cpu 的话不用考虑 cuda。
引用与感谢
本文地址:https://blog.csdn.net/l1l1l1l/article/details/107659020
下一篇: Python减少循环层次和缩进的技巧分析