安装 vai_q_tensorflow - 3.5 简体中文

Vitis AI 用户指南 (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 简体中文

vai_q_tensorflow 有两种安装方法:

使用 Docker 容器安装

Vitis AI 为量化工具(包括 vai_q_tensorflow)提供了 Docker 容器。运行容器后,请激活 Conda 环境 Vitis AI -tensorflow

[docker] $ conda activate vitis-ai-tensorflow
如果 Docker 容器内有 package-tensorflow 补丁包,则请在 Docker 容器内安装 Vitis AI-tensorflow 补丁包:
[docker] $ sudo env CONDA_PREFIX=/opt/vitis_ai/conda/envs/vitis-ai-tensorflow/ PATH=/opt/vitis_ai/conda/bin:$PATH conda install patch_package.tar.bz2

使用源代码安装

vai_q_tensorflow 是 AMD 维护的插件工具,适用于 TensorFlow 1.15。它在 Vitis_AI_Quantizer 中开源。要构建 vai_q_tensorflow,请运行以下命令:
[host] $ sh build.sh