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 xilinx_ai_optimizer@xilinx.com to access the Vitis AI Optimizer installation package and license. Follow the installation steps to install the pre-requirements and Vitis AI Optimizer.