vai_q_tensorflow2 は次の 3 つの方法でインストールできます。
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 パッケージのソース コードからインストール
vai_q_tensorflow2 は、TensorFlow モデル最適化ツールキットのフォークです。これは Vitis_AI_Quantizer 内のオープンソースです。vai_q_tensorflow2 を構築するには、次のコマンドを実行します。
[host] $ sh build.sh
[host] $ pip install pkgs/*.whl
Conda パッケージのソース コードからインストール
重要: これには 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