Vitis AI 运行时包和 Vitis AI Library 样本与模型都已编译到预构建的 Vitis AI 开发板镜像中。因此,如果目标当前使用的是 Vitis AI 预构建镜像,那么您无需安装 Vitis AI 运行时包和模型包。但是,您仍可在定制目标上或者在官方镜像上遵循如下步骤安装模型:
- 对于每个模型,都有一个 yaml 文件用于描述有关模型的详细信息。此 YAML 文件包含适用于各 AMD 目标开发板的下载链接。选择您的模型和所需平台并下载。
- 使用
scp
搭配以下命令将下载的文件复制到目标。$ scp <model>.tar.gz root@IP_OF_BOARD:~/
如果目标开发板连接到互联网,您也可以使用
wget
命令将模型直接下载到开发板。 - 登录到目标开发板(使用 ssh 或串行端口)并安装模型程序包。
- 如果在目标上 /usr/share/vitis_ai_library/model 文件夹不存在,请首先创建此文件夹。
# mkdir -p /usr/share/vitis_ai_library/models
- 在目标开发板上安装模型。
# tar -xzvf <model>.tar.gz -C /usr/share/vitis_ai_library/models
默认情况下,这些模型位于目标上的 /usr/share/vitis_ai_library/models 目录中。