运行 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

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

Tensorflow Module Not Found Error when running MNIST TPU

tensorflow

tpu