Docker Setup#
The following steps outline the process for pulling and running the Docker container required for the Vitis AI model quantization and compilation flows.
Pull the Docker image on a system with Docker installed. For installation instructions, see Install Docker Engine on Ubuntu and Linux Post-Installation Steps for Docker Engine. You can also contact your system administrator.
docker pull amdih/vitis-ai:versal-2ve-release_v6.2_0612
Note
Docker image pulling can take a few minutes.
Verify the Docker image details (“REPOSITORY”, “IMAGE ID” and “TAG”):
docker imagesThe output appears in the following format:
REPOSITORY
TAG
IMAGE ID
CREATED
SIZE
amdih/vitis-ai
versal-2ve-release_v6.2_0612
8cd54102c274
xx hours ago
31.2GB
Get the
AIEMLv2buildbuild tools license (node locked/floating) as described in Obtaining License. Copy the node locked license file to your workspace at<yourLicenseDir>.Start a Docker container and mount the required directories:
docker run --ulimit stack=-1:-1 -it --rm --network host -v <yourLicenseDir>:/usr/licenses -v <yourDir>:/host_mount amdih/vitis-ai:versal-2ve-release_v6.2_0612 bash
This command uses a node-locked license. If you have a floating license, see Docker Run Command with Floating License for example commands to run docker with Floating license.
Note
Use the
-voption to map the host directory (<yourDir>) to the container directory (/host_mount).Large models can require a larger stack size. Use
--ulimit stack=-1:-1to remove the stack limit.