Pytorch环境配置Tensorboard对训练进行实时可视化

这里主要介绍一下自己踩过的坑。

我自己的pytorch版本:

torchvision 0.6.1
pytorch 1.5.1 【GPU版本】

在该版本下,训练过程如果需要可视化,需要安装两个东西。推荐使用pip安装。

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorboardx
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorboard

从清华源镜像完成安装完成后,tensorboard 版本为 2.3.0,tensorboardx 版本为 2.1。

这里注意,如果报错了建议卸载重新安装;注意一定使用清华源镜像;注意在一个网速好的环境不然会出现“read time out” 的错误。

安装完成后,如何进行基础操作的理解与测试呢?这里推荐
https://github.com/lanpa/tensorboardX

这里有一个demo程序。复制粘贴下来。
对照Github作者写的一些基础包,都安装好。【这些包很小,安装很快】
运行该demo。
更改conda路径至当前程序所在文件夹,然后在conda的cmd栏键入“tensorboard --logdir runs”命令,按照提示在浏览器中打开Tensorboard即可。
Pytorch环境配置Tensorboard对训练进行实时可视化
关于一些基础操作可以见:
https://www.pytorchtutorial.com/pytorch-builtin-tensorboard/

后记:
Tensorboard装了一下午。有的人说Tensorflow,tensorboard,tensorboardX都需要,有的人说只需要其中一个。这就很晕。加上网速不好,下个包有时候就报一堆红错,心情很差,搞了半天。

最近压力比较大,实在是头秃的八月。总感觉自己孤立无援,整个实验室也就我这么一个人在做这个方向。整个人陷入不断的怀疑与自我否定,精神状态也不是很好。

phd is a journey…

Pytorch环境配置Tensorboard对训练进行实时可视化

相关文章:

  • 2021-12-13
  • 2022-12-23
  • 2021-11-26
  • 2021-08-15
  • 2021-12-22
  • 2021-12-07
  • 2022-12-23
  • 2021-12-05
猜你喜欢
  • 2022-12-23
  • 2021-09-17
  • 2022-12-23
  • 2021-04-16
  • 2021-12-04
  • 2021-05-17
  • 2021-10-21
相关资源
相似解决方案