Ubuntu下,Pytorch可视化工具tensorboardX

今天在用pytocrch框架运行代码的时候,出现了错误MoudleNotFoundError:No module named ‘tensorboardX’,一时情急,赶紧百度,有人说是路径不对,但是也有人说是缺少模块(果真如此),在经过亲身试错后,发现这是因为没有安装tensorflow引起的,tensorboardX是在tensorboard的基础上做出来,需要tensorflow支持。
Ubuntu下,Pytorch可视化工具tensorboardX_第1张图片

1. tensorboadX

  • Pytorch也能用tensorboard啦,虽然Pytorch还有visdom可视工具,但是对于从tensorflow转来或者双修党来说,还是tensorboard用起来习惯点咯。
  • tensorboardX是在tensorboard的基础上做出来,需要tensorflow支持
  • 想在pytorch中进行数据可视化,感觉tensorboard的语句有点麻烦,所以想要用tensorboardX。

2. 安装

  • 先安装tensorflow: conda install tensorflow
    Ubuntu下,Pytorch可视化工具tensorboardX_第2张图片
  • 再安装tensorboardX: conda install tensorboardX
    当我们安装好tensorflow的时候,要进一步去安装tensorboardX,要不然依旧会出现错误
    Ubuntu下,Pytorch可视化工具tensorboardX_第3张图片
    可以看到,但运行train.py的文件时,出现了另外的一个错误(无地自容,了,菜到极致了),错误出现在py文件中的第18行,而不是原先的第14行
    参考
    参考2

你可能感兴趣的:(pytorch,Anaconda)