pytorch - TensorBoard:教程 Pytorch:模块 'tensorflow._api.v2.io.gfile' 没有属性 'get_filesystem'

Question

我在运行有关带有嵌入的 Pytorch 和 TensorBoard 的教程时遇到问题 https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html

我收到了这条消息，

AttributeError                            Traceback (most recent call last)
<ipython-input-10-e0404d94b4cd> in <module>()
     22 writer.add_embedding(features,
     23                     metadata=class_labels,
---> 24                     label_img=images.unsqueeze(1))
     25 writer.close()

/usr/local/lib/python3.6/dist-packages/torch/utils/tensorboard/writer.py in add_embedding(self, mat, metadata, label_img, global_step, tag, metadata_header)
    779         save_path = os.path.join(self._get_file_writer().get_logdir(), subdir)
    780 
--> 781         fs = tf.io.gfile.get_filesystem(save_path)
    782         if fs.exists(save_path):
    783             if fs.isdir(save_path):

我可以想象投影仪

Answer 1

试试这个

import tensorflow as tf
import tensorboard as tb
tf.io.gfile = tb.compat.tensorflow_stub.io.gfile