There are two ways to obtain the Vitis AI optimizer:
- Docker Image
-
Vitis AI provides a Docker environment for the
Vitis AI optimizer. The following conda
environments are available in the Docker:
- vitis-ai-optimizer_pytorch
- vitis-ai-optimizer_tensorflow
- vitis-ai-optimizer_tensorflow2
- vitis-ai-optimizer_caffe
- vitis-ai-optimizer_darknet
All requirements are ready in these environments. If you have a license, you can run the Vitis AI pruner directly in the Docker.
- 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.