Skip to content

Commit 9ce25ea

Browse files
Weiming Zhaoxuhongyao
Weiming Zhao
authored andcommitted
[Docker] Update scripts to differentiate runtime/devel images
1 parent 7194f6a commit 9ce25ea

File tree

3 files changed

+12
-7
lines changed

3 files changed

+12
-7
lines changed

utils/docker/Dockerfile

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# Build this image: docker build -t halo:[version] .
22

33
ARG BASE_IMAGE
4-
FROM ${BASE_IMAGE}
4+
FROM ${BASE_IMAGE}
55

66
# Redeclare the argument
77
ARG BASE_IMAGE
@@ -73,7 +73,7 @@ RUN apt-get update -y && \
7373
ninja-build \
7474
libopencv-core-dev \
7575
libopencv-highgui-dev \
76-
libopencv-videoio-dev \
76+
libopencv-videoio-dev \
7777
libmpc-dev \
7878
libmpfr-dev \
7979
libgmp-dev\
@@ -226,7 +226,7 @@ RUN cd /tmp && \
226226
git clone --depth=1 https://github.com/google/flatbuffers.git -b v1.12.0 && \
227227
cd flatbuffers && \
228228
cmake -G "Unix Makefiles" -DCMAKE_BUILD_TYPE=Release -DFLATBUFFERS_BUILD_SHAREDLIB=ON && make -j && make install && \
229-
rm -fr /tmp/flatbuffers
229+
rm -fr /tmp/flatbuffers
230230

231231

232232
RUN mkdir -p /var/run/sshd

utils/docker/build_image.sh

+6-2
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,14 @@
11
#!/bin/bash
22

3+
VER="latest"
4+
FLAVOR="devel"
5+
NAMESPACE="registry-intl.us-west-1.aliyuncs.com/computation"
6+
37
base_image_gpu="nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04"
48
base_image_cpu="ubuntu:18.04"
59

610
docker build --build-arg BASE_IMAGE=${base_image_cpu} \
7-
-t latest-x86_64-ubuntu18.04 .
11+
-t $NAMESPACE/halo:$VER-$FLAVOR-x86_64-ubuntu18.04 .
812

913
docker build --build-arg BASE_IMAGE=${base_image_gpu} \
10-
-t latest-cuda10.0-cudnn7-ubuntu18.04 .
14+
-t $NAMESPACE/halo:$VER-$FLAVOR-cuda10.0-cudnn7-ubuntu18.04 .

utils/docker/start_docker_cpu.sh

+3-2
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,16 @@
11
#!/bin/bash -xe
22

33
VER="latest"
4+
FLAVOR="devel"
45
NAMESPACE="registry-intl.us-west-1.aliyuncs.com/computation"
56

67
if [[ "$0" =~ "start_docker_gpu.sh" ]]; then
7-
IMAGE="$NAMESPACE/halo:$VER-cuda10.0-cudnn7-ubuntu18.04"
8+
IMAGE="$NAMESPACE/halo:$VER-$FLAVOR-cuda10.0-cudnn7-ubuntu18.04"
89
CONTAINER_NAME=$USER.halo-$VER-GPU
910
docker_run_flag="--runtime=nvidia"
1011
else
1112
use_gpu=0
12-
IMAGE="$NAMESPACE/halo:$VER-x86_64-ubuntu18.04"
13+
IMAGE="$NAMESPACE/halo:$VER-$FLAVOR-x86_64-ubuntu18.04"
1314
CONTAINER_NAME=$USER.halo-$VER-CPU
1415
docker_run_flag=""
1516
fi

0 commit comments

Comments
 (0)