Installation - 1.4 English

Vitis AI Optimizer User Guide (UG1333)

Document ID
Release Date
1.4 English

The following are the two ways to obtain the Vitis AI Optimizer:

Docker Image
Vitis AI provides a docker environment for the Optimizer. In the docker image, there are three optimizer related conda environments: vitis-ai-optimizer_tensorflow, vitis-ai-optimizer_caffe, and vitis-ai-optimizer_darknet. All requirements are ready in these environments. CUDA and cuDNN versions in the docker are CUDA 10.0 and cuDNN 7.6.5. After getting a license, you can run the VAI pruner directly in the docker.
Note: The optimizer for PyTorch is not in the docker image and can only be installed using the conda package.
Conda Packages
Conda packages are also available for Ubuntu 18.04. Contact to access the Vitis AI Optimizer installation package and license. Follow the installation steps to install the pre-requirements and Vitis AI Optimizer.