解决Pytorch中调用tensorboard报错 AttributeError: module ‘tensorflow‘ has no attribute ‘gfile‘

问题描述

在执行下面命令时出现报错：

from torch.utils.tensorboard import SummaryWriter

tb_writer = SummaryWriter(osp.join(opt.savepath, "logdir"))

报错内容：

AttributeError: module 'tensorflow._api.v1.io' has no attribute 'gfile'

问题分析

出现这个问题的根本原因在于pytorch调了Tensorflow，最后由Tensorflow报出的错误，tensorflow的新版本与旧版本的不兼容。

经查看，我的pytorch版本为1.6，tensorboard版本为1.15.0，tensorflow版本为1.12.0

问题解决

升级tensorflow版本为2.0版本即可，执行下面命令会自动升级tensorflow版本和tensorboard版本：

pip install tensorflow==2.0

完。