Colab导入错误：无法导入名称squeeze_or_expand_dimensions

Question

以下是使用TensorFlow Hub进行TFLite迁移学习教程，第一步是

!pip install -q -U "tensorflow-gpu=2.0.0b1"

设置：

import os

import matplotlib.pylab as plt
import numpy as np

import tensorflow as tf
import tensorflow_hub as hub

print("Version: ", tf.__version__)
print("Eager mode: ", tf.executing_eagerly())
print("Hub version: ", hub.__version__)
print("GPU is", "available" if tf.test.is_gpu_available() else "NOT AVAILABLE")

出现错误:ImportError:尝试后无法导入名称“export_saved_model”

%tensorflow_version 2.x
!pip uninstall -y tensorflow
!pip install tensorflow-gpu==1.14.0

根据此答案，引发了另一个错误：ImportError：无法导入名称“squareze_or_expand_dimensions”

另外，我尝试了！pip安装-升级tenorflow，并提出了另一个错误…

Answer 1

不要运行 ！pip install -q -U “tensorflow-gpu==2.0.0b1”。

在“运行时”菜单下，“更改运行时类型”为“GPU”和“工厂重置运行时”。

import os

import matplotlib.pylab as plt
import numpy as np

import tensorflow as tf
import tensorflow_hub as hub

print("Version: ", tf.__version__)
print("Eager mode: ", tf.executing_eagerly())
print("Hub version: ", hub.__version__)
print("GPU is", "available" if tf.test.is_gpu_available() else "NOT AVAILABLE")

输出：

Version:  2.2.0
Eager mode:  True
Hub version:  0.8.0
GPU is available