From a3595bf6c803173359fc4583239e12174533e123 Mon Sep 17 00:00:00 2001 From: Boyka_Occam Date: Thu, 16 Sep 2021 01:17:39 +0000 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0dockerfile=E5=92=8C=E5=AE=B9?= =?UTF-8?q?=E5=99=A8=E5=90=AF=E5=8A=A8=E8=84=9A=E6=9C=AC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit update official/nlp/pangu_alpha/docker_start.sh. --- official/nlp/pangu_alpha/Dockerfile | 5 ++++ official/nlp/pangu_alpha/docker_start.sh | 36 ++++++++++++++++++++++++ 2 files changed, 41 insertions(+) create mode 100644 official/nlp/pangu_alpha/Dockerfile create mode 100644 official/nlp/pangu_alpha/docker_start.sh diff --git a/official/nlp/pangu_alpha/Dockerfile b/official/nlp/pangu_alpha/Dockerfile new file mode 100644 index 000000000..053bf80cd --- /dev/null +++ b/official/nlp/pangu_alpha/Dockerfile @@ -0,0 +1,5 @@ +ARG FROM_IMAGE_NAME +FROM ${FROM_IMAGE_NAME} + +COPY requirements.txt . +RUN pip3.7 install -r requirements.txt \ No newline at end of file diff --git a/official/nlp/pangu_alpha/docker_start.sh b/official/nlp/pangu_alpha/docker_start.sh new file mode 100644 index 000000000..703fff6d5 --- /dev/null +++ b/official/nlp/pangu_alpha/docker_start.sh @@ -0,0 +1,36 @@ +#!/bin/bash +# Copyright 2021 Huawei Technologies Co., Ltd +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# ============================================================================ + +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 \ + -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \ + -v /usr/local/Ascend/add-ons/:/usr/local/Ascend/add-ons/ \ + -v ${model_dir}:${model_dir} \ + -v ${data_dir}:${data_dir} \ + -v /root/ascend/log:/root/ascend/log ${docker_image} /bin/bash -- Gitee