运行 MNIST TPU 时未找到 Tensorflow 模块错误答案

【问题标题】：Tensorflow Module Not Found Error when running MNIST TPU运行 MNIST TPU 时未找到 Tensorflow 模块错误
【发布时间】：2020-05-20 12:54:18
【问题描述】：

我是 tensorflow 的新手，正在尝试按照此处的教程 https://cloud.google.com/tpu/docs/quickstart 运行 MNIST TPU 模型。

我收到了来自

的错误

  python3 mnist_main.py \
  --tpu=$TPU_NAME \
  --model_dir=$MODEL_DIR \
  --data_dir=$DATA_DIR \
  --train_epochs=10 \
  --distribution_strategy=tpu \
  --download

错误信息

  Traceback (most recent call last):
  File "mnist_main.py", line 31, in <module>
    from official.vision.image_classification.resnet import common
  File "/usr/share/models/official/vision/image_classification/resnet/common.py", line 26, in <module>
    import tensorflow_model_optimization as tfmot
ModuleNotFoundError: No module named 'tensorflow_model_optimization'

有解决的办法吗？谢谢。

【问题讨论】：

标签： tensorflow tpu

【解决方案1】：

您需要使用命令行（或终端）安装它。

pip install --user --upgrade tensorflow-model-optimization

【讨论】：

我用的是 Linux，好像 pip 在 linix 上不起作用。抱歉，我应该指定我的操作系统
sudo apt-get install pip
这是您需要学习的真正技能，它可以解决您遇到的大多数错误。从您刚刚遇到的两个问题中，这表明您需要先用 Google 搜索您的问题，如果这样做了，请学习如何更有效地使用 Google。 Pip 在 Linux 上可用。如果您收到类似 pip not found 的消息，您的第一反应应该是谷歌“如何安装 pip”甚至“找不到 pip”，然后您就会找到答案。这并不意味着居高临下。我想帮助您更有效地解决问题。
我在所有地方的 colab 都面临这个问题！该软件包已安装在我的系统中，我已将其用于其他东西，但我不得不将“实验”层与新模型一起使用，jupyter 不同意，因此我移至 colab，其他东西也可以使用，但现在我有来自 colab 的“ModuleNotFoundError: No module named 'tensorflow_model_optimization”。有什么建议吗？

【解决方案2】：

在 Google Colab 中，如果您使用 qkeras 运行笔记本，则需要执行以下操作：

!git clone https://github.com/google/qkeras.git 
import sys 
sys.path.append('qkeras') 
!pip install git+https://github.com/keras-team/keras-tuner.git 
!pip install tensorflow_model_optimization 
import tensorflow_model_optimization as tfmot

【讨论】：