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

Question

我是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'

有没有办法解决这个问题？谢谢。

Answer 1

您需要使用命令行(或终端)安装它。

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

Answer 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