环境说明:win7,python3.6.5,cuda9.0
版本说明:使用pytorch1.0.1和tensorboard1.9.0,tensorflow-gpu1.9.0
在使用tensorboard之前需要首先安装tensorflow,我安装的是GPU版本的。
然后参考两个入门
基础版:https://www.jianshu.com/p/46eb3004beca
增强版:https://blog.csdn.net/qq_36653505/article/details/84727253
上面的代码可以跑通。
踩过的坑:
主要是两个,(1)logdir的路径最好是详细路径,是event文件所在的路径
(2)开头使用的pytorch0.4.0,但是在对model进行可视化时,总是报OperatorExportTypes的错误,通过查阅源码发现是torch的版本不支持OperatorExportTypes,所以将0.4.0版本的torch卸载改为1.0.1版本的torch,错误解决。