首先建立一个文件夹log
运行一下脚本
from tensorboardX import SummaryWriter
writer = SummaryWriter("log")
for i in range(100):
writer.add_scalar("a",i,global_step = i)
writer.add_scalar("b",i ** 2,global_step = i)
writer.close()
在终端输入如下命令
(pytorch_gpu) E:\pytorch>cd log
系统找不到指定的路径。
(pytorch_gpu) E:\pytorch>cd 003
(pytorch_gpu) E:\pytorch\003>cd log
(pytorch_gpu) E:\pytorch\003\log>tensorboard --logdir ./
2021-01-28 23:28:35.357052: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully
opened dynamic library cudart64_101.dll
Serving TensorBoard on localhost; to expose to the network, use a proxy or pass --bind_all
TensorBoard 2.2.2 at http://localhost:6006/ (Press CTRL+C to quit)
打开
http://localhost:6006/#scalars
success.