欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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