Tensorboard-pytorch简单入门

安装

pip install tensorboardX  (对应tensorflow的版本)

或者源文件编译:

git clone https://github.com/lanpa/tensorboardX && cd tensorboardX && python setup.py install

测试

clone上面的链接后,使用里面的例子进行测试:

python examples/demo.py

启动Tensorboard

在服务器端输入:

tensorboard --logdir=‘logs’ --port==10043

在本地浏览器上:输入GPU服务器的IP:端口号 ,之后可以看到:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_18649781/article/details/89847872