pytorch - tensorboard colab tensorflow._api.v1.io.gfile' 没有属性 'get_filesystem

Question

我正在尝试在 colab 上使用张量板。我设法使它工作，但不适用于所有命令。 add_graph 和 add_scalar 有效，但是当我尝试运行 add_embedding 时，出现以下错误:

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

这是相关的代码(我认为)；

import os
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter(log_dir ="logs" )

images, labels = select_n_random(trainset.data, trainset.targets)
images = torch.from_numpy(images)
labels = torch.from_numpy(np.array(labels))
class_labels = [classes[lab] for lab in labels]

# log embeddings
features = images.reshape((-1,32*32*3))
writer.add_embedding(features,metadata=class_labels) #, label_img=images.unsqueeze(1))

完整的错误是:

/tensorflow-1.15.0/python3.6/tensorflow_core/python/util/module_wrapper.py in __getattr__(self, name)
    191   def __getattr__(self, name):
    192     try:
--> 193       attr = getattr(self._tfmw_wrapped_module, name)
    194     except AttributeError:
    195       if not self._tfmw_public_apis:

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

使用

tensorflow-1.15.0(尝试安装 2.0 但有不同的问题)

Python 3.6.9

火炬 1.4.0

tensorboard 2.1.1(也尝试过 1.15.0 但同样的问题)

我也尝试使用“魔法”命令:

%load_ext tensorboard
%tensorboard --logdir logs

但我无法让它以这种方式工作(其他问题)。

任何建议如何使它工作？

Answer 1

对我来说，这解决了问题:

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