@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