先把tensorflow和pytorch安装好
之后在pytorch环境里
pip install tensorboardX
将
with SummaryWriter(comment=‘LeNet’) as w:
w.add_graph(model, (varInput,))
添加到模型和输入的后面，类似这样注意参数第一个是模型，第二个是输入
然后将数据集和测试集变小一点，迭代次数设为一代，跑一次直到程序跑完，文件夹下会出现一个runs文件类似这样
之后用cmd或者anaconda打开带有之前安装tensorboardx的环境（由于我的pytorch和tensorflow在不同的环境里所以要用两个环境）cd到runs文件夹 （路径里不能有汉字）
tensorboard --logdir runs
像这样结果
然后去谷歌打开这个网址就行了或者打开http://localhost:6006也是可以的