有人让 Tensorflow 分布式培训笔记本工作吗？

Question

当我运行这个从 Tensorflows 文档站点链接的笔记本时，

https://colab.research.google.com/github/tensorflow/docs/blob/r2.0rc/site/en/tutorials/distribute/training_loops.ipynb#scrollTo=7x7s5iYAYSGD

我收到以下错误：

模块“tensorflow._api.v1.nn”没有属性“compute_average_loss”

如果有人可以让它工作，你在 Google Colab 中使用什么设置？ Python 3/2、GPU、Tensorflow版本等

谢谢

Answer 1

github代码是从今年5月开始的：

https://github.com/tensorflow/docs/commits/r2.0rc/site/en/tutorials/distribute/training_loops.ipynb

当 colab 加载 tf 1.14 时：

from __future__ import absolute_import, division, print_function, unicode_literals

# Import TensorFlow
import tensorflow as tf

# Helper libraries
import numpy as np
import os

print(tf.__version__)

1.14.0

所以你最好的选择是使用 tf 2.0 运行程序。

希望能帮助到你：）