-
Notifications
You must be signed in to change notification settings - Fork 4
/
build.sh
executable file
·58 lines (45 loc) · 2.01 KB
/
build.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
57
58
#!/bin/bash
IMAGE_BASE_NAME="pure/python"
set -e
cat readme_header.md >README_.md
echo -e "## Supported tags\n" >>README_.md
for py in "3.8" "3.7" "3.6"; do
echo -e "### Python $py\n" >>README_.md
for cuda in "10.2" "10.1" "10.0" "9.0"; do
echo -e "#### CUDA ${cuda}\n" >>README_.md
IMAGE_PREFIX="${IMAGE_BASE_NAME}:${py}"
echo "Building ${IMAGE_PREFIX}-cuda${cuda}-base"
docker build \
--quiet \
--build-arg PY="${py}" \
--tag "${IMAGE_PREFIX}-cuda${cuda}-base" \
"${cuda}/base"
echo
echo "Building ${IMAGE_PREFIX}-cuda${cuda}-runtime"
docker build \
--quiet \
--build-arg IMAGE_PREFIX="${IMAGE_PREFIX}" \
--build-arg CUDA="${cuda}" \
--tag "${IMAGE_PREFIX}-cuda${cuda}-runtime" \
"${cuda}/runtime"
echo
echo "Building ${IMAGE_PREFIX}-cuda${cuda}-cudnn7-runtime"
docker build \
--quiet \
--build-arg IMAGE_PREFIX="${IMAGE_PREFIX}" \
--build-arg CUDA="${cuda}" \
--tag "${IMAGE_PREFIX}-cuda${cuda}-cudnn7-runtime" \
"${cuda}/runtime/cudnn7"
echo
echo "Pushing images to repository"
docker push "${IMAGE_PREFIX}-cuda${cuda}-base"
docker push "${IMAGE_PREFIX}-cuda${cuda}-runtime"
docker push "${IMAGE_PREFIX}-cuda${cuda}-cudnn7-runtime"
echo
echo "- [\`${IMAGE_PREFIX}-cuda${cuda}-base\` (*${cuda}/base/Dockerfile*)](https://github.com/cicdteam/python-cuda/blob/master/${cuda}/base/Dockerfile)" >>README_.md
echo "- [\`${IMAGE_PREFIX}-cuda${cuda}-runtime\` (*${cuda}/runtime/Dockerfile*)](https://github.com/cicdteam/python-cuda/blob/master/${cuda}/runtime/Dockerfile)" >>README_.md
echo "- [\`${IMAGE_PREFIX}-cuda${cuda}-cudnn7-runtime\` (*${cuda}/runtime/cudnn7/Dockerfile*)](https://github.com/cicdteam/python-cuda/blob/master/${cuda}/runtime/cudnn7/Dockerfile)" >>README_.md
echo "" >>README_.md
done
done
mv -f README_.md README.md