Installation - 2.5 English

Vitis AI Optimizer User Guide (UG1333)

Document ID
Release Date
2.5 English

Vitis AI provides a Docker environment for the Vitis AI Optimizer. To build and run the Docker image, refer to the "GPU docker" section in the documentation. The following conda environments are available in the Docker image:

  • vitis-ai-optimizer_pytorch
  • vitis-ai-optimizer_tensorflow
  • vitis-ai-optimizer_tensorflow2

The Docker encapsulates the required tools and libraries necessary for pruning in these frameworks. If you have a license, you can run the Vitis AI Optimizer directly in the Docker.