Skip to content

Latest commit

 

History

History
37 lines (22 loc) · 709 Bytes

load_run_docker.md

File metadata and controls

37 lines (22 loc) · 709 Bytes

Docker Images Are Hosted on Docker Hub HERE

The GPU docker has been tested with GPU machines with Docker 19.03.1, NVIDIA driver 410.xx (to work with CUDA 10.0) and nvidia-docker 2.2.2.

CPU tools docker

# Load the CPU image from dockerhub

./docker_run.sh xilinx/vitis-ai-cpu:latest

Or 

# Build the CPU image and load it

cd Vitis-AI/docker
./docker_build_cpu.sh
cd ..
./docker_run.sh xilinx/vitis-ai-cpu:latest

GPU tools docker

# GPU Image must be built by user. 

cd Vitis-AI/docker
./docker_build_gpu.sh

# After the GPU image is built, load it

cd ..
./docker_run.sh xilinx/vitis-ai-gpu:latest