文档
注册

原生Docker挂载vNPU

原生Docker场景主要为使用npu-smi工具创建vNPU后,将vNPU挂载到容器。

使用前提

按照规划参见创建vNPU完成对vNPU的创建。

使用方法

用户通过npu-smi工具创建vNPU后,在拉起容器时执行以下命令将vNPU挂载至容器中。以下命令表示用户在拉起容器时,挂载虚拟芯片ID为100的芯片。

docker run -it \
--device=/dev/vdavinci100:/dev/davinci100 \
--device=/dev/davinci_manager \
--device=/dev/devmm_svm \
--device=/dev/hisi_hdc \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/sbin/npu-smi:/usr/local/sbin/npu-smi \
-v /home:/home \
-v /usr/local/Ascend/driver/lib64/common:/usr/local/Ascend/driver/lib64/common \
-v /usr/local/Ascend/driver/lib64/driver:/usr/local/Ascend/driver/lib64/driver \
-v /etc/ascend_install.info:/etc/ascend_install.info \
-v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
docker_image_id  /bin/bash

参数

参数说明

--device

表示映射的设备,vdavinci100是申请的虚拟设备,需要根据实际设备名称修改。其他保持不变。注意映射到容器中需要重命名为davinci100,对应的id保持不变,仅将vdavinci修改为davinci。其它的保持不变。

-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi

将宿主机npu-smi工具“/usr/local/bin/npu-smi”和“/usr/local/sbin/npu-smi”挂载到容器中,请根据实际情况修改。

-v /usr/local/sbin/npu-smi:/usr/local/sbin/npu-smi

-v /usr/local/Ascend/driver/lib64/common:/usr/local/Ascend/driver/lib64/common

将宿主机目录“/usr/local/Ascend/driver/lib64/common”和“/usr/local/Ascend/driver/lib64/driver”挂载到容器中。请根据driver的驱动SO所在路径修改。

-v /usr/local/Ascend/driver/lib64/driver:/usr/local/Ascend/driver/lib64/driver

-v /etc/ascend_install.info:/etc/ascend_install.info

将宿主机安装信息文件“/etc/ascend_install.info”挂载到容器中。

-v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info

将宿主机版本信息文件“/usr/local/Ascend/driver/version.info”挂载到容器中,请根据实际情况修改。

docker_image_id

容器镜像ID,请替换为实际容器镜像ID。

执行之后即可启动容器,并将创建的虚拟设备映射到容器中。

容器启动之后,执行以下命令查看当前Docker容器中可以使用的davinci设备,如果有davinci设备则表示虚拟设备成功映射到容器中。

ls /dev/ | grep davinci*
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词