The following are the two ways to obtain Vitis AI Optimizer:
- Docker image. Vitis AI provide docker environment for 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 reqirements are ready in these environments. CUDA and cuDNN versions in docker are CUDA 10.0 and cuDNN 7.6.5. After getting a license, you can run the VAI pruner directly in the docker.
- Conda packages are also available for Ubuntu 18.04. Contact Xilinx Marketing/sales/FAE to get the conda packages. Follow the installation steps to install pre-reqirements and Vitis AI Optimizer.