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.
- Follow the Post-installation steps for Linux to ensure that 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