【问题记录】AttributeError: module ‘tensorflow‘ has no attribute ‘io‘

原创已于 2022-07-06 17:29:41 修改 · 2.2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #pytorch

于 2022-07-01 19:38:10 首次发布

本文解决了一个关于PyTorch中TensorBoard使用的兼容性问题，具体表现为尝试使用TensorBoard时出现AttributeError错误。解决方案是安装特定版本的TensorFlow以确保与PyTorch中的TensorBoard组件兼容。

部署运行你感兴趣的模型镜像

from torch.utils.tensorboard import SummaryWriter

遇到报错：

File "D:\anaconda3\envs\pytorch_py37\lib\site-packages\torch\utils\tensorboard\writer.py", line 220, in __init__
    self._get_file_writer()
  File "D:\anaconda3\envs\pytorch_py37\lib\site-packages\torch\utils\tensorboard\writer.py", line 251, in _get_file_writer
    self.flush_secs, self.filename_suffix)
  File "D:\anaconda3\envs\pytorch_py37\lib\site-packages\torch\utils\tensorboard\writer.py", line 61, in __init__
    log_dir, max_queue, flush_secs, filename_suffix)
  File "D:\anaconda3\envs\pytorch_py37\lib\site-packages\tensorboard\summary\writer\event_file_writer.py", line 72, in __init__
    tf.io.gfile.makedirs(logdir)
  File "D:\anaconda3\envs\pytorch_py37\lib\site-packages\tensorboard\lazy.py", line 65, in __getattr__
    return getattr(load_once(self), attr_name)
AttributeError: module 'tensorflow' has no attribute 'io'

虽然Pytorch直接使用的是tensorboard，但还是需要配合tensorflow使用，解决方法：