如何保存 ternsorflow keras 模型？

Question

我正在尝试使用以下摘要保存 tensorflow keras 模型：

Model: "sequential_2" etc.

使用以下命令：

model.save('my_model.h5')

我收到以下错误（在 Jupyter Notebook 中）：

ImportError: `save_model` requires h5py.

所以，我安装了h5py，使用conda conda install h5py安装后，我在Jupyter Notebook中得到了h5py的版本（我试图保存模型的地方）：

h5py.__version__
'2.8.0'

尽管如此，我还是遇到了同样的错误。 尽管如此，我还是手动导入了 h5py。

import h5py
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers
filename = 'model.h5' 
model.save(filename)

---------------------------------------------------------------------------
**ImportError**                               Traceback (most recent call last)
<ipython-input-54-9160eee81fe6> in <module>
      5 from tensorflow.keras import layers
      6 filename = 'model.h5'
----> 7 model.save(filename)

~/anaconda3/lib/python3.7/site-packages/tensorflow_core/python/keras/engine/network.py in save(self, filepath, overwrite, include_optimizer, save_format, signatures)

~/anaconda3/lib/python3.7/site-packages/tensorflow_core/python/keras/saving/save.py in save_model(model, filepath, overwrite, include_optimizer, save_format, signatures)

~/anaconda3/lib/python3.7/site-packages/tensorflow_core/python/keras/saving/hdf5_format.py in save_model_to_hdf5(model, filepath, overwrite, include_optimizer)

**ImportError:** save_model requires h5py.

Answer 1

您是否尝试过直接安装h5py ？
如果你有一个现有的 Python 安装（例如 python.org 下载，或者你的操作系统附带的那个），那么在 Intel 计算机上的 Windows、macOS/OSX 和 Linux 上，可以通过 pip 从PyPI：

$ pip install h5py

在Windows 中：
如果失败，您可以尝试安装 cython
pip install h5py
pip install cython

在Linux/Ubuntu 中：
如果失败，那么您可能还需要 libhdf5 安装 libhdf5 然后安装 h5py
1. sudo pip install cython
2. sudo apt-get install libhdf5-dev
3. sudo pip install h5py

注意：如果上述安装不起作用，请尝试卸载 h5py，然后重新安装并重新启动 jupyter notebook 或 anaconda。