You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
1.1 KiB
30 lines
1.1 KiB
#!/bin/bash
|
|
|
|
docker_image=$1
|
|
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 \
|
|
--privileged \
|
|
-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 \
|
|
-v /var/log/npu/:/usr/slog ${docker_image} \
|
|
/bin/bash
|