加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
docker_start.sh 967 Bytes
一键复制 编辑 原始数据 按行查看 历史
zhaoqiang 提交于 2020-12-21 16:58 . 增加docker启动文件,修改readme
#!/usr/bin/env bash
docker_image=$1 #接受第一个参数作为docker_image
data_dir=$2 #接受第二个参数作为训练数据集路径
model_dir=$3 #接受第三个参数作为模型执行路径
docker run -it --ipc=host \
--device=/dev/davinci0 --device=/dev/davinci1 --device=/dev/davinci2 --device=/dev/davinci3 --device=/dev/davinci4 --device=/dev/davinci5 --device=/dev/davinci6 --device=/dev/davinci7 \
--device=/dev/davinci_manager --device=/dev/devmm_svm --device=/dev/hisi_hdc \
-v /usr/local/Ascend/driver:/usr/local/Ascend/driver -v /usr/local/Ascend/add-ons/:/usr/local/Ascend/add-ons/ \
-v ${data_dir}:${data_dir} \
-v ${model_dir}:${model_dir} \
-v /var/log/npu/conf/slog/slog.conf:/var/log/npu/conf/slog/slog.conf \
-v /var/log/npu/slog/:/var/log/npu/slog -v /var/log/npu/profiling/:/var/log/npu/profiling \
-v /var/log/npu/dump/:/var/log/npu/dump ${docker_image} \
/bin/bash
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化