在 colab 上运行pytorch使用tensorboard的官方教程时遇到问题:
module ‘tensorflow._api.v2.io.gfile’ has no attribute ‘get_filesystem’
问题代码如下:
def select_n_random(data,labels,n=100):
assert len(data)==len(labels)
perm = torch.randperm(len(data))
return data[perm][:n],labels[perm][:n]
images,labels=select_n_random(trainset.data,trainset.targets)
class_labels = [classes[lab] for lab in labels]
features = images.view(-1,28*28)
writer.add_embedding(features,metadata=class_labels,label_img=images.unsqueeze(1))
writer.close()
网上说是因为环境中同时安装了 tensorflow 和 pytorch ,建议卸载 tensorflow ,博主当时的表情:
把 colab 的 tensorflow 卸载……先看看有没有别的方法吧。
在Stack Overflow上找到了一个解决方法!运行下列代码:
import tensorflow as tf
import tensorboard as tb
tf.io.gfile = tb.compat.tensorflow_stub.io.gfile
再运行之前的代码,亲测没有报错了。
原解决方法链接:Stack Overflow解决方法