首先在服务器安装tensorboard,直接使用pip安装即可。
pip install tensorboard
在代码中的具体使用如下：
导入tensorboard
from torch.utils.tensorboard import SummaryWriter
在代码中加入
log_dir = os.path.join('loss', 'train_G_loss')
train_writer = SummaryWriter(log_dir=log_dir)
log_dir = os.path.join('loss', 'train_D_loss')
test_writer = SummaryWriter(log_dir=log_dir)
train_writer.add_scalar('loss', train_G_loss, epoch)
test_writer.add_scalar('loss', train_D_loss, epoch)
其中‘loss'为保存的文件名，train_G_loss和train_D_loss是绘制的两种损失。
在本地可视化服务器上的tensorboard分为如下三步：
1.首先在服务器上运行tensorboard的程序，指定端口号（除6006以外的端口，未使用的，可行）。
tensorboard --logdir=loss --port=6123
将loss替换为想要查看的文件路径，“ 6123 ”替换为想要在服务器上监听的端口。
2.使用ssh命令，在windows的cmd里连接服务器，并实现服务器端口与本地端口的映射。
ssh 用户名@IP -L 8123：127。0.0.1：6123
ssh -p命令用来连接远程服务器，-L命令用来将服务器监听端口映射到本地端口。因此，将用户名，IP替换为想要连接的目标服务器的信息，将“ 8123 ”替换为想要在本地映射的端口。
3.
本地浏览器内查看tensorboard
http://localhost:8123/
