当前位置： 移动技术网 > IT编程>脚本编程>Python > pytorch笔记--tensorboardX出错：不能显示graph网络结构

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

2020年07月30日  | 移动技术网IT编程  | 我要评论
本机环境  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

您可能感兴趣的文章:

如您对本文有疑问或者有任何想说的，请点击进行留言回复，万千网友为您解惑！

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网