pytorch的tensorboard显示遇到的坑
此文中记录使用tensorboard显示pytorch的一些坑
按照pytorch官方的教程遇到了无数的坑,特记录如下,方便日后查阅学习。因为是通过回忆记录,所以没有描述具体细节。
最终运行环境如下:
python 3.7.6
pytorch 1.2.0
cudatoolkit 10.0.130
tensorflow-gpu 2.0.0
tensorboard 2.0.2
tensorboardx 2.0
1. 需要搭配安装的package
官方教程上导入tensorboard的方法为:from torch.utils.tensorboard import SummaryWriter
但是会报错:ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed. This should be available in 1.14 or above.
但这个时候你去检查tensorboard的版本很有可能是高于1.14的,解决办法为使用tensorboardX:
pytorch以外,还需要同时安装tensorboardX, tensorflow-gpu(cpu版本也不行),tensorboard,conda安装或者pip安装都可以
2. pytorch的安装
通过pytorch官网的命令安装安装up to date版本的torch一直报错,安装了old version的torch:
‘conda install