pytorch+tensorboard远程可视化

安装

pip install tensorboard


使用

流程就是在代码中把值写入log文件&＃xff0c;然后用tensorboard读取就行

from torch.utils.tensorboard import SummaryWriter
import torch
import numpy as np
writer &＃61; SummaryWriter(logsdir)
x &＃61; np.arange(0, 5, 0.1)
x &＃61; torch.from_numpy(x)for i in x:yi &＃61; torch.exp(i)writer.add_scalar(&＃39;exp&＃39;, yi, i)


注意SummaryWriter(logsdir) 其中的logsdir是要保存的路径&＃xff0c; writer.add_scalar(tag, scalar_value, global_step&＃61;None, walltime&＃61;None)这里的tag指定变量名字&＃xff0c;scalar_value是变量的值&＃xff0c;global_step为x轴坐标。

保存如下

配置

1. 在本地计算机上使用SSH 并将远程服务器的端口映射到本地的计算机。

ssh -L 6006:127.0.0.1:6006 username&＃64;server_ip -p server_port


2. 在远程服务器开启tensorboard

tensorboard --logdir logsdir --port&＃61;6006


这里的logsdir 就是上文的保存路径

运行后弹出如下链接

粘贴到本地服务器即可看到

Linux中&＃xff0c;ERROR: TensorBoard could not bind to port 6006, it was already in use问题解决&＃xff1a;

终端输入&＃xff1a;

lsof -i:6006


这时候在终端会发现&＃xff1a;

COMMAND PID USER FD TYPE DEVICE SIZE/OFF NODE NAME
tensorboa PID .......


然后关闭PID

sudo kill -9 PID