vai_q_tensorflow2 のインストール - 2.5 日本語

Docker コンテナーを使用してインストール

Vitis AI は、vai_q_tensorflow を含む量子化ツールの Docker コンテナーを提供します。コンテナーの実行後、Conda 環境 vitis-ai-tensorflow2 をアクティベートします。

conda activate vitis-ai-tensorflow2

パッチ パッケージがある場合、Docker コンテナー内の vitis-ai-tensorflow2 パッチ パッケージをインストールします。

# [optional]
$ 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 を構築するには、次のコマンドを実行します。

$ sh build.sh
$ pip install pkgs/*.whl

Conda パッケージのソース コードからインストール

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