docker run -it -d --net=host --shm-size=1g \ --name <container-name> \ -w /home \ --device=/dev/davinci0 \ --device=/dev/davinci1 \ --device=/dev/davinci2 \ --device=/dev/davinci3 \ --device=/dev/davinci_manager \ --device=/dev/hisi_hdc \ --device=/dev/devmm_svm \ -v /usr/local/Ascend/driver:/usr/local/Ascend/driver:ro \ -v /usr/local/dcmi:/usr/local/dcmi:ro \ -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi:ro \ -v /usr/local/sbin/:/usr/local/sbin:ro \ -v /path-to-weights:/path-to-weights:ro \ mindie-turbo:800I-A2-py311-Openeuler24.03-aarch64 bash
“mindie-turbo:800I-A2-py311-Openeuler24.03-aarch64”为镜像名称,可根据实际情况修改。
参数 |
参数说明 |
---|---|
--name |
设置容器名称。 |
--device |
表示映射的设备,可以挂载一个或者多个设备。 需要挂载的设备如下:
说明:
可根据以下命令查询device个数及名称方式,根据需要绑定设备,修改上面命令中的"--device=****"。 ll /dev/ | grep davinci |
-v /usr/local/Ascend/driver:/usr/local/Ascend/driver:ro |
将宿主机目录“/usr/local/Ascend/driver”挂载到容器,请根据驱动所在实际路径修改。 |
-v /usr/local/sbin:/usr/local/sbin:ro |
将宿主机工具“/usr/local/sbin/”以只读模式挂载到容器中,请根据实际情况修改。 |
-v /path-to-weights:/path-to-weights:ro |
设定权重挂载的路径,需要根据用户的情况修改。 说明:
请将权重文件和数据集文件同时放置于该路径下。 |
1 | docker exec -it <container-name> /bin/bash |