/
run.prod.sh
executable file
·56 lines (47 loc) · 1.58 KB
/
run.prod.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#!/bin/bash
SCRIPTDIR=$(dirname "${BASH_SOURCE[0]}")
WORKDIR=$SCRIPTDIR/../..
if [ "$1" != "cpu" ] && [ "$1" != "gpu" ]; then
echo "missing compute option CPU or GPU"
echo "example usage: ./scripts/build.sh cpu"
exit 1
fi
if [[ -v "$MODEL_PATH" ]]; then
echo "MODEL_PATH environment varibale not set"
echo "first run: export MODEL_PATH=/path/to/cntk/model/"
exit 1
fi
if [ "$1" == 'gpu' ]; then
# Fix for GPU machine w/ nvidia setup
if [ -f '/usr/bin/nvidia-smi' ]; then
mv /usr/bin/nvidia-cuda-mps-control /usr/bin/nvidia-cuda-mps-control1
mv /usr/bin/nvidia-cuda-mps-server /usr/bin/nvidia-cuda-mps-server1
mv /usr/bin/nvidia-debugdump /usr/bin/nvidia-debugdump1
mv /usr/bin/nvidia-persistenced /usr/bin/nvidia-persistenced1
mv /usr/bin/nvidia-smi /usr/bin/nvidia-smi1
mv /usr/bin/nvidia-xconfig /usr/bin/nvidia-xconfig1
fi
docker run --rm -itd \
--runtime=nvidia \
-v $MODEL_PATH:/workdir/model \
-e PORT=80 \
-e NVIDIA_VISIBLE_DEVICES=all \
-p 3000:80 \
--name vcntkprod-gpu \
user1m/vott-reviewer-cntk:$1
if [ -f '/usr/bin/nvidia-smi1' ]; then
mv /usr/bin/nvidia-cuda-mps-control1 /usr/bin/nvidia-cuda-mps-control
mv /usr/bin/nvidia-cuda-mps-server1 /usr/bin/nvidia-cuda-mps-server
mv /usr/bin/nvidia-debugdump1 /usr/bin/nvidia-debugdump
mv /usr/bin/nvidia-persistenced1 /usr/bin/nvidia-persistenced
mv /usr/bin/nvidia-smi1 /usr/bin/nvidia-smi
mv /usr/bin/nvidia-xconfig1 /usr/bin/nvidia-xconfig
fi
else
docker run --rm -itd \
-v $MODEL_PATH:/workdir/model \
-e PORT=80 \
-p 3000:80 \
--name vcntkprod-cpu \
user1m/vott-reviewer-cntk:$1
fi