Pytorch下tensorboard的安装与配置

Pytorch下tensorboard的安装与配置

1.tensorboard简介

tensorboard是tensorflow开发的一款绘图插件，它可以绘制网络的图像，可以绘制训练时的 Loss ,Accuracy等参数指标，tensorboard现在已经支持在pytorch中使用，使用方法参考pytorch文档，链接如下：

https://pytorch.org/docs/stable/tensorboard.html

2.安装

• 在线安装

  使用pip安装命令，在windows命令窗口中输入以下安装命令，安装任一版本的tensorflow都可以，注意要将tensorboard与tensorflow安装在pytorch所在的换环境内。

pip install tensorflow==1.15

pip install tensorboard

• 离线安装

  进入.whl文件所在的网站，此处以清华镜像源为例，下载对应版本到本地。注意，要对应本地环境的python版本。

https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple/

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-EXplb2mZ-1631883196119)(C:\Users\16618\AppData\Roaming\Typora\typora-user-images\image-20210917202816504.png)]

​ 在cmd中，将目录cd到文件所在的位置，使用Pip安装。

pip install tensorflow-1.9.0-cp36-cp36m-win_amd64.whl

pip install tensorboard

3.验证安装

打开pycharm，在python Console中输入以下代码：

from torch.utils.tensorboard import SummaryWriter

如不提示错误，则安装成功，若提示以下错误：

TensorBoard logging requires TensorBoard version 1.15 or above

进入以下链接，安装tensorborad的nightly版本

https://pypi.org/project/tb-nightly/