tensorboard使用_colab打不开tensorboard的解决办法

如果你在谷歌云端的colab上用pytorch框架打开tensorboard,发生以下打不开的情况,其实只需要改动一点点代码
tensorboard使用_colab打不开tensorboard的解决办法_第1张图片
原代码(点击网站无法显示):
我是想将logs函数显示出来,范围是x轴100以内

from torch.utils.tensorboard import SummaryWriter
writer=SummaryWriter('logs')
for i in range(100):
  writer.add_scalar("y=x",i,i)
!tensorboard --logdir=logs --port=6006   #可以指定端口,也可以不指定

以下是可以出来结果的执行代码,分为两步

1、先安装tensorboardX包

! pip install tensorboardX
%load_ext tensorboard

2、将上文中的!改为%,这样不用点击网页,而是直接显示在运行结果处

from torch.utils.tensorboard import SummaryWriter
writer=SummaryWriter('logs')
for i in range(100):
  writer.add_scalar("y=x",i,i)
%tensorboard --logdir=logs --port=6006   # 改动的地方,将!改为%

结果运行如下:
tensorboard使用_colab打不开tensorboard的解决办法_第2张图片

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