Browse Source

!15897 add docker start to deeplabv3

From: @jiangzg001
Reviewed-by: @c_34,@wuxuejian
Signed-off-by: @c_34
pull/15897/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
8e126e3254
13 changed files with 37 additions and 4 deletions
  1. +0
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/space_to_depth_base.h
  2. +0
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/space_to_batch_fp32.h
  3. +0
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/gather_parameter.h
  4. +0
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/gatherNd_int8.h
  5. +0
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/gather_int8.h
  6. +0
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/leaky_relu_int8.h
  7. +0
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/sigmoid_int8.h
  8. +0
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/squeeze_int8.h
  9. +0
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/unsqueeze_int8.h
  10. +0
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/space_to_depth_parameter.h
  11. +6
    -0
      model_zoo/official/cv/deeplabv3/Dockerfile
  12. +2
    -4
      model_zoo/official/cv/deeplabv3/requirements.txt
  13. +29
    -0
      model_zoo/official/cv/deeplabv3/scripts/docker_start.sh

+ 0
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/space_to_depth_base.h View File


+ 0
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/space_to_batch_fp32.h View File


+ 0
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/gather_parameter.h View File


+ 0
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/gatherNd_int8.h View File


+ 0
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/gather_int8.h View File


+ 0
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/leaky_relu_int8.h View File


+ 0
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/sigmoid_int8.h View File


+ 0
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/squeeze_int8.h View File


+ 0
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/unsqueeze_int8.h View File


+ 0
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/space_to_depth_parameter.h View File


+ 6
- 0
model_zoo/official/cv/deeplabv3/Dockerfile View File

@@ -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

+ 2
- 4
model_zoo/official/cv/deeplabv3/requirements.txt View File

@@ -1,4 +1,2 @@
mindspore
numpy
Pillow
python-opencv
opencv-python >= 4.1.2.30
pillow >= 6.2.0

+ 29
- 0
model_zoo/official/cv/deeplabv3/scripts/docker_start.sh View File

@@ -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…
Cancel
Save