Vitis AI

Загрузка и запуск Docker

Образы Docker размещены в Docker Hub ЗДЕСЬ

Образ GPU был протестирован на машинах с графическим процессором с Docker 19.03.1, драйвером NVIDIA версии 410.xx (для работы с CUDA 10.0) и nvidia-docker версии 2.2.2.

Теги образа Docker на Docker Hub

Существует соответствующая связь между Vitis AI и версией образа Docker, обычно вам нужно найти соответствующий образ Docker для использования. Среди них самый простой способ — напрямую использовать предварительно созданный образ Docker на Docker Hub. Просто нужно выполнить следующую команду. Для последней версии Vitis AI <x.y.z> можно просто заменить на latest.

./docker_run.sh xilinx/vitis-ai-cpu:<x.y.z>

Соответствие версий между Vitis AI и образом Docker показано в следующей таблице.

Использование CPU tools docker

Просто запустите образ Docker CPU с помощью следующей команды. Если указанный образ Docker не существует, команда автоматически загрузит его из Docker Hub.

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

Если вы не хотите загружать образ из Docker Hub, а собирать его из файла Docker. Затем вам нужно выполнить команду следующим образом, чтобы собрать образ Docker.

cd Vitis-AI/docker
./docker_build_cpu.sh
cd ..

Затем вы можете использовать ту же команду, как показано ниже, для загрузки и запуска собранного образа Docker.

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

Использование GPU tools docker

Для образа GPU Docker необходимо собрать его из файла Docker, и предварительно созданного образа нет. Команда сборки выглядит следующим образом.

cd Vitis-AI/docker
./docker_build_gpu.sh
cd ..

Затем вы можете использовать команду, как показано ниже, чтобы загрузить и запустить собранный образ Docker.

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