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