pytorch在远程服务器上调用tensorboard

tensorboard作为一个基于web的可视化界面,由于其界面美观操作简单的特点,受到用户的广泛好评。最早tensorboard只支持tensorflow框架的模型,但是随着用户对tensorboard需求的提升与其他深度框架的崛起,出现了一批能够将tensorboard应用于其他框架的开源工具。
本文是笔者在将tensorboard用于pytorch框架中的收集到的资料,包括了跨框架的tensorboard调用与服务器的tensorboard调用两个部分。

跨框架的tensorboard调用

跨框架的tensorboard调用有很多的方式,常见的有tensorboardX,tensorboard_logger等方式,在pytorch1.1之后的版本中,更是提供了tensorboard优秀的原生支持,只需要一句简单的「from torch.utils.tensorboard import SummaryWriter」指令就可以调用。
tensorboardX相关教程:
https://www.jianshu.com/p/46eb3004beca
tensorboard_logger相关教程:
https://blog.csdn.net/wen_fei/article/details/82977630
由于采用的是比较低的pytorch版本,所以我无法调用原生的tensorboard,而且采用tensorboard主要喜欢其画图的几个功能,所以我采用的是一个简单的脚本,包含了scalar,image,histo三个绘图功能。
脚本相关教程:
https://blog.csdn.net/qq_16234613/article/details/79835119

服务器上的tensorboard调用

由于我是在服务器上运行的代码,所以tensorboard的显示结果也会在服务器上,如果需要查看服务器上的信息,就需要建立一个隧道,将服务器的端口映射到本地上。
使用MobaXterm配置ssh隧道教程:
https://blog.csdn.net/u010520552/article/details/80823953
在服务器使用TensorBoard的方法:
https://blog.csdn.net/sinat_35512245/article/details/82960937
XShell上建立隧道相关教程:
https://blog.csdn.net/index20001/article/details/82871634
如何在服务器上远程使用tensorboard?
https://www.jianshu.com/p/9a0ecef29025

以下是踩的坑:

  • 建立隧道的过程是在主机上,而非服务器。
  • 若是调用tensorboard命令时,找不到tensorboard,就需要找到tensorboard的安装地址,tensorboardX也会存在类似问题。相关博客:https://blog.csdn.net/appleml/article/details/76976006

你可能感兴趣的:(深度框架,tensorboard,pytorch)