Issue with tensorboard in pytorch

1

I am trying to use tensorboard with pytorch (1.2.0+cu92). When calling writer = SummaryWriter(), I get the following error:
AttributeError: module 'tensorflow.io' has no attribute 'gfile'

How can I fix this? The following few lines would reproduce the error.

import torch
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter()
2 Likes