PyTorch1.6.0使用Tensorboard

需要的环境

Tensorflow-gpu == 1.13.1
Tensorboard == 1.15.0

先安装Tensorflow-gpu == 1.13.1, 然后卸载Tensorboard == 1.13.1,在自己安装Tensorboard == 1.15.0

sudo pip install -i https://pypi.douban.com/simple tensorflow-gpu==1.13.1
sudo pip uninstall tensorboard
sudo pip install -i https://pypi.douban.com/simple tensorboard==1.15.0

如果之前有安装tensorflow-gpu,则可以先更新tensorflow-gpu的版本

sudo pip install --upgrade -i https://pypi.douban.com/simple tensorflow-gpu==1.13.1

说明：

PyTorch1.6.0只允许Tensorboard == 1.15.0及以上，但是Tensorflow-gpu1.0只更新到1.14.0，而且Tensorflow-gpu == 1.14.0与Tensorboard==1.15.0不兼容，所以用Tensorflow-gpu == 1.13.1