pytorch笔记--tensorboardX出错：不能显示graph网络结构

本机环境

  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/caiguanhong/article/details/106924606