AttributeError：模块“tensorflow._api.v2.config.experimental”没有属性“set_lms_enabled”答案

【问题标题】：AttributeError: module 'tensorflow._api.v2.config.experimental' has no attribute 'set_lms_enabled'AttributeError：模块“tensorflow._api.v2.config.experimental”没有属性“set_lms_enabled”
【发布时间】：2020-10-03 08:56:06
【问题描述】：

我想在我的深度学习项目中使用 LMS，我为此使用了代码：

import tensorflow as tf
tf.config.experimental.set_lms_enabled(True)

来自这个IBM GitHub link。但我得到了提到的错误。

AttributeError: module 'tensorflow._api.v2.config.experimental' has no attribute 'set_lms_enabled'

我在使用 tensorflow 2.2 的 colab notebook 中尝试过。我应该降级吗？我该如何解决这个问题？

【问题讨论】：

标签： tensorflow google-colaboratory

【解决方案1】：

根据TensorFlow Large Model Support github page：

TFLMS 内置于 tensorflow-gpu conda 包中，因此当您从 WML CE 安装支持 GPU 的 TensorFlow 时，默认情况下会安装它。

所以您需要安装 anaconda 并通过 conda 包管理器从 WML CE 安装 TensorFlow。要安装 anaconda，请通过以下链接访问 anaconda 官方文档页面： https://docs.anaconda.com/anaconda/install/

【讨论】：