The Vitis AI 3.5 release uses containers to distribute the AI software. The release consists of the following components:
- Tools container
- Public GitHub for examples (https://github.com/Xilinx/Vitis-AI)
- Vitis AI Model Zoo (https://github.com/Xilinx/Vitis-AI/tree/v3.5/model_zoo)
Tools Container
The tools container consists of the following:
- Containers distributed through Docker Hub: https://hub.docker.com/r/xilinx/
- Unified compiler flow includes:
- Compiler flow for DPUCZDX8G (Edge)
- Compiler flow for DPUCVDX8G (Edge)
- Compiler flow for DPUCV2DX8G (Edge and Data Center)
- Pre-built conda environment to run frameworks:
-
conda activate vitis-ai-tensorflow
for TensorFlow-based flows -
conda activate vitis-ai-tensorflow2
for TensorFlow2-based flows -
conda activate vitis-ai-pytorch
for PyTorch-based flowsNote: For WeGO workflow in PyTorch, please activate the following conda environment:conda activate vitis-ai-wego-torch
-
- Versal Runtime tools
Note: Caffe has been deprecated from
AMD Vitis™ AI 2.5. For information on Caffe, see
Vitis AI 2.0 User Guide.