File tree 3 files changed +12
-7
lines changed
3 files changed +12
-7
lines changed Original file line number Diff line number Diff line change 1
1
# Build this image: docker build -t halo:[version] .
2
2
3
3
ARG BASE_IMAGE
4
- FROM ${BASE_IMAGE}
4
+ FROM ${BASE_IMAGE}
5
5
6
6
# Redeclare the argument
7
7
ARG BASE_IMAGE
@@ -73,7 +73,7 @@ RUN apt-get update -y && \
73
73
ninja-build \
74
74
libopencv-core-dev \
75
75
libopencv-highgui-dev \
76
- libopencv-videoio-dev \
76
+ libopencv-videoio-dev \
77
77
libmpc-dev \
78
78
libmpfr-dev \
79
79
libgmp-dev\
@@ -226,7 +226,7 @@ RUN cd /tmp && \
226
226
git clone --depth=1 https://github.com/google/flatbuffers.git -b v1.12.0 && \
227
227
cd flatbuffers && \
228
228
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
230
230
231
231
232
232
RUN mkdir -p /var/run/sshd
Original file line number Diff line number Diff line change 1
1
#! /bin/bash
2
2
3
+ VER=" latest"
4
+ FLAVOR=" devel"
5
+ NAMESPACE=" registry-intl.us-west-1.aliyuncs.com/computation"
6
+
3
7
base_image_gpu=" nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04"
4
8
base_image_cpu=" ubuntu:18.04"
5
9
6
10
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 .
8
12
9
13
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 .
Original file line number Diff line number Diff line change 1
1
#! /bin/bash -xe
2
2
3
3
VER=" latest"
4
+ FLAVOR=" devel"
4
5
NAMESPACE=" registry-intl.us-west-1.aliyuncs.com/computation"
5
6
6
7
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"
8
9
CONTAINER_NAME=$USER .halo-$VER -GPU
9
10
docker_run_flag=" --runtime=nvidia"
10
11
else
11
12
use_gpu=0
12
- IMAGE=" $NAMESPACE /halo:$VER -x86_64-ubuntu18.04"
13
+ IMAGE=" $NAMESPACE /halo:$VER -$FLAVOR - x86_64-ubuntu18.04"
13
14
CONTAINER_NAME=$USER .halo-$VER -CPU
14
15
docker_run_flag=" "
15
16
fi
You can’t perform that action at this time.
0 commit comments