Blog Post #2

A tutorial for setting up Vitis AI

Vitis AI is a comprehensive Integrated Development Environment (IDE) from Xilinx for AI inference on hardware platform, such as ZCU102/ZCU104 FPGA board. It supports different frameworks like PyTorch and TensorFlow from the user application. Comparing to the method using High-level synthesis (HLS), Vitis AI provides a simpler way to optimize the Neural Network (NN) model and incresae the efficiency for AI acceleration.

This tutorial will show you how to set up the Vitis AI on Ubuntu 22.04 environment.

Step1:
Open terminal and run the following cmd to install Docker first.
sudo apt install docker.io

Step2:
Initialize Docker by cmd.
sudo docker run hello-world

Step3:
Clone the repository of Vitis AI from Github.
git clone https://github.com/Xilinx/Vitis-AI

Step4:
Navigate to the directory of Vitis AI and run it.
cd Vitis-AI
sudo ./docker_run.sh xilinx/vitis-ai
If you see something like permission denied for running docker_run.sh, then run the following cmd to give permission.
chmod +x docker_run.sh

Step5:
Done!
If you are using Pytorch, you may use the following cmd to activate the environment
conda activate vitis-ai-pytorch
You should see something similar as below:

Hope you enjoy this :)

Reference:
https://www.xilinx.com/video/software/setting-up-vitis-ai-environment.html