#!/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