时间戳

首页 > 解决方案 > 有人让 Tensorflow 分布式培训笔记本工作吗？

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

问题描述

当我运行这个从 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 版本等。

谢谢

标签： tensorflowgoogle-colaboratory

解决方案

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 运行程序。

希望能帮助到你 ：）

推荐阅读