pytorch安装tensorboardX

  1. tensorboardX是pytorch的可视化工具
  2. tensorboard作为Tensorflow中强大的可视化工具,已经被广泛使用
    但针对其他框架,例如Pytorch,之前一直没有这么好的可视化工具可用,好在目前Pytorch也可以支持Tensorboard了,那就是通过使用tensorboardX,真是Pytorcher的福利!
  3. 因为tensorboardX是对tensorboard进行了封装后,开放出来使用,所以必须先安装tensorboard, 再安装tensorboardX!!
    必须先安装tensorboard再安装tensorboardX!!!否则会报错
  4. 安装步骤
    step1 安装tensorflow,安装tensorflow会自动安装tensorboard
    可以参考另外一篇文章安装tensorflow
pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

pytorch安装tensorboardX_第1张图片
step2 确定tensorboard安装好之后安装tensorboardX

pip install tensorboardX

pytorch安装tensorboardX_第2张图片
step3 tensorboardX已经全部安装好,在Terminal中输入tensorboard --logdir=路径,点击链接即可
在这里插入图片描述
pytorch安装tensorboardX_第3张图片
参考文献:pytorch可视化tensorboard

你可能感兴趣的:(pytorch,pytorch,深度学习)