Pytorch环境配置Tensorboard对训练进行实时可视化
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 i
原创
2020-08-13 21:29:24 ·
1539 阅读 ·
2 评论