下载pytorch1.7.0以上的tensorboard及使用
一、下载
- 建议下载的时候使用加速地址,如清华源
pip install tensorboard
- 下面是下载时的效果图,pytorh1.7.0版本后,tensorboard已经嵌入到pytorch中了,但是需要下载非常多的包,目前不知道是否可以和TensorFlow框架中的一起使用,博主没有亲自尝试。
二、使用
- 下面为tensorboard使用的命令,也可以同时打开多个tensorboard,只需要修改port端口号就可以,还是比较简单的,就是tensorboard没有将之前训练轮次蓄起来的功能,但是是可以通过一定的逻辑和技巧实现的,这个博主会在后面的文章有介绍。
from torch.utils.tensorboard import SummaryWriter
summary_writer = SummaryWriter("logs")
summary_writer.close()
"""
[命令]: tensorboard --logdir=logs --port=6006
"""