本文共 266 字,大约阅读时间需要 1 分钟。
PyTorch 从 1.2.0 版本开始,正式自带内置的 Tensorboard 支持了,我们可以不再依赖第三方工具来进行可视化。所以首先要保证你的pytorch必须是1.2.0以上版本的。
安装
打开终端,输入
conda activate rootpip install tensorboard -i https://pypi.douban.com/simple/
测试
打开jupyter输入下面代码,不出错就说明安装成功。
from torch.utils.tensorboard import SummaryWriter
转载地址:http://cgjq.baihongyu.com/