Vitis AI Pruner - 2.0 English

Vitis AI Optimizer User Guide (UG1333)

Document ID

UG1333

Release Date

2022-01-20

Version

2.0 English

Note: For the Docker image, skip this section.

To install the Vitis AI pruner, first install Conda and then install the Conda package for your framework.

Install Conda

For more information, see the Conda installation guide.

vai_optimizer_tensorflow

$ tar xzvf vai_optimizer_tensorflow.tar.gz
$ conda install vai_optimizer_tensorflow_gpu -c file://$(pwd)/vai-bld -c conda-forge/label/gcc7 -c conda-forge

vai_optimizer_tensorflow2

$ tar xzvf vai_optimizer_tensorflow2.tar.gz
$ conda install vai_optimizer_tf2 -c file://$(pwd)/vai-bld -c conda-forge

vai_optimizer_pytorch

$ tar xzvf vai_optimizer_pytorch.tar.gz
$ conda install vai_optimizer_pytorch_gpu -c file://$(pwd)/vai-bld -c pytorch

vai_optimizer_caffe

$ tar xzvf vai_optimizer_caffe.tar.gz
$ conda install vai_optimizer_caffe_gpu -c file://$(pwd)/vai-bld -c conda-forge/label/gcc7 -c conda-forge

vai_optimizer_darknet

$ tar xzvf vai_optimizer_darknet.tar.gz
$ conda install vai_optimizer_darknet_gpu -c file://$(pwd)/vai-bld