安装模型卡工具包

使用 pip 安装

模型卡工具包托管在 PyPI 上,需要 Python 3.7 或更高版本。对于大多数用例,使用 pip 安装是最佳方法。

安装基本、框架无关的软件包

pip install model-card-toolkit

如果您要为 TensorFlow 模型生成模型卡,请安装可选的 TensorFlow 依赖项以使用模型卡工具包的 TensorFlow 实用程序

pip install model-card-toolkit[tensorflow]

从 pip 20.3 开始,您可能需要在运行时追加 --use-deprecated=legacy-resolver 标志。

从源代码安装

如果您想为该项目贡献代码或使用最新功能,从源代码安装是最佳选择。按照 贡献指南 中的说明了解如何设置本地开发环境。按照以下步骤从源代码安装 wheel。

在您可以从源代码安装模型卡工具包之前,您需要安装 Bazel>=2.0.0,它支持 protobuf 存根代码生成。

首先,克隆 GitHub 存储库

git clone https://github.com/tensorflow/model-card-toolkit.git
cd model-card-toolkit

从源代码构建 pip 包

python setup.py sdist bdist_wheel

最后,安装您本地构建的包

pip install --upgrade ./dist/model_card_toolkit-*.whl

解决安装问题

如果您在安装模型卡工具包时遇到问题,请 提交问题,其中包含有关您的操作系统版本、Python 版本、pip 版本和本地安装的软件包的详细信息。您可以使用 pip freeze 找到您本地安装的软件包。