pytorch使用tensorflowboardx进行观察

@pytorch使用tensorflowboardx进行观察

安装pytorch后跑神经网络代码

两个机器，或者一个足以，使用anaconda安装了不同的环境，
第一个是在CPU上跑的pytorch，conda虚拟环境就叫pytorch，我是复制了一个这样的环境然后进行的操作先试试水
另一个是GPU版的，conda虚拟环境叫pytorch_gpu，
（复制环境的命令，conda create -n py_gpu_tsor --clone pytorch_gpu）
我再pytorch_gpu里面conda list 观察发现已经安装了tensorboardx 1.4，然后查阅资料说要想运行tenserboardx要先安装tensorflow，当然，GPU版要安装GPU版的tensorflow，于是conda install tensorflow-gpu搞定
CPU版的同理，只不过安装的是CPU版的，直接 conda install tensorflow， CPU下的tensorboardx反而是1.9，tensorboardx好像都是在安装pytorch时直接安装的。

tensorboardx的使用方法，参考https://blog.csdn.net/qq_36653505/article/details/84727253
简要来说，现在代码中加入需要观察的变量，通过

from tensorboardX import SummaryWriter
writer = SummaryWriter(log_dir='logs')

其实logs是存放目录。
运行命令简单，在正确的conda环境和log所在目录下终端运行 tensorboard --logdir logs