将 tensorflow gpu 模型转换为 cpu 进行推理

Question

我有 TensorFlow 预训练权重用于推理。当我安装 tensorflow-gpu==2.2.0 时，我得到了结果。但是当我安装 tensorflow-cpu==2.2.0 时，我得到了 nan。我想在cpu上在线托管模型，那我该怎么办。 这就是我加载模型的方式

model = Model()
model.load_weights('weights.hdf5')
model.predict()