|
|
|
@@ -0,0 +1,29 @@ |
|
|
|
#!/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 |