安装 vai_q_tensorflow2 - 3.5 简体中文

Vitis AI 用户指南 (UG1414)

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

可通过以下 3 种方式来安装 vai_q_tensorflow2:

使用 Docker 容器安装

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

[docker] $ conda activate vitis-ai-tensorflow2

如有补丁包,请在 Docker 容器内部安装 Vitis AI-tensorflow2 补丁包。

# [optional]
[docker] $ sudo env CONDA_PREFIX=/opt/vitis_ai/conda/envs/vitis-ai-tensorflow2/ PATH=/opt/vitis_ai/conda/bin:$PATH conda install patch_package.tar.bz2

使用 Wheel Package 从源代码安装

vai_q_tensorflow2 是 TensorFlow 模型最优化工具集的复刻。它在 Vitis_AI_Quantizer 中开源。要构建 vai_q_tensorflow2,请运行以下命令:

[host] $ sh build.sh
[host] $ pip install pkgs/*.whl

使用 Conda Package 从源代码安装

重要: 这需要 Anaconda。

# CPU-only version
[host] $ conda build vai_q_tensorflow2_cpu_feedstock --output-folder ./conda_pkg/
# GPU version
[host] $ conda build vai_q_tensorflow2_gpu_feedstock --output-folder ./conda_pkg/
# Install conda package on your machine
[host] $ conda install --use-local ./conda_pkg/linux-64/*.tar.bz2