tensorboard是一个可视化工具,我在pytorch中使用,首先要先装这个包,利用下面代码:
pip install tensorboard
在利用测试代码看看效果:
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs")
# writer.add_image()
# y = x**2
for i in range(100):
writer.add_scalar("y = x**2", i ** 2, i)
writer.close()
然后在有logs文件夹下,按住shift+右键,点击“在此处打开Powershell窗口”,输入cmd,再输入代码激活环境
conda activate pytorch1.5
再输入代码,打开tensorboard
tensorboard --logdir=logs --port=6007
这时打开谷歌浏览器,输入
http://localhost:6007/
这样就可以看到我们画的图了