The following two options are available
for installing the containers with the Vitis AI tools and resources.
- Pre-built containers on Docker Hub: xilinx/vitis-ai
- Build containers locally with Docker recipes: Docker Recipes
Use the following steps for installation:
- Install Docker - If Docker not installed on your machine yet.
- Ensure your linux user is in the group docker
- Clone the Vitis AI repository to obtain the examples, reference code, and
scripts.
git clone --recurse-submodules https://github.com/Xilinx/Vitis-AI cd Vitis-AI
-
Run Docker Container
- Run the CPU image from docker
hub
docker pull xilinx/vitis-ai:latest ./docker_run.sh xilinx/vitis-ai
- Build the CPU image locally and run
it
cd docker ./docker_build_cpu.sh # After build finished cd .. ./docker_run.sh xilinx/vitis-ai-cpu:latest
- Build the GPU image locally and run
it
cd docker ./docker_build_gpu.sh # After build finished cd .. ./docker_run.sh xilinx/vitis-ai-gpu:latest
- Run the CPU image from docker
hub