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 https://github.com/Xilinx/Vitis-AI#installation 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 image 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 image.