forked from mindspore-Ecosystem/mindspore
add docker start to deeplabv3
This commit is contained in:
parent
786a20cd4f
commit
b197b1a427
|
@ -0,0 +1,6 @@
|
||||||
|
ARG FROM_IMAGE_NAME
|
||||||
|
FROM ${FROM_IMAGE_NAME}
|
||||||
|
|
||||||
|
RUN apt install libgl1-mesa-glx -y
|
||||||
|
COPY requirements.txt .
|
||||||
|
RUN pip3.7 install -r requirements.txt
|
|
@ -1,4 +1,2 @@
|
||||||
mindspore
|
opencv-python >= 4.1.2.30
|
||||||
numpy
|
pillow >= 6.2.0
|
||||||
Pillow
|
|
||||||
python-opencv
|
|
||||||
|
|
|
@ -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
|
Loading…
Reference in New Issue