-
Notifications
You must be signed in to change notification settings - Fork 5
137 lines (134 loc) · 5.9 KB
/
build_ml.yml
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
name: build_icetray_ml
env:
ICETRAY_VERSION: v1.8.1
on:
workflow_dispatch:
push:
branches: [ master ]
paths:
- 'icecube_ml/Dockerfile*'
jobs:
docker:
runs-on: ubuntu-latest
steps:
- name: Get current date
id: date
run: echo "current_date=$(date +'%Y-%m-%d')" >> $GITHUB_ENV
-
name: Set up Docker Buildx
uses: docker/setup-buildx-action@v2
- name: Free Disk Space (Ubuntu)
uses: jlumbroso/free-disk-space@main
with:
# this might remove tools that are actually needed,
# if set to "true" but frees about 6 GB
tool-cache: true
# all of these default to true, but feel free to set to
# "false" if necessary for your workflow
android: true
dotnet: true
haskell: true
large-packages: true
swap-storage: true
-
name: Login to Docker Hub
uses: docker/login-action@v2
with:
username: ${{ secrets.DOCKERHUB_USERNAME }}
password: ${{ secrets.DOCKERHUB_TOKEN }}
# -
# name: Build and push icetray ml cuda 11.6 for 20 04
# uses: docker/build-push-action@v3
# with:
# push: true
# file: icecube_ml/Dockerfile_ubuntu_20.04_cuda11.6
# build-args: |
# VERSION=${{ env.ICETRAY_VERSION }}
# tags: >
# icecube/icetray-ml:icetray-cuda11.6-${{ env.ICETRAY_VERSION }}-ubuntu20.04,
# icecube/icetray-ml:icetray-cuda11.6-current-ubuntu20.04,
# icecube/icetray-ml:icetray-cuda11.6-${{ env.ICETRAY_VERSION }}-ubuntu20.04-${{ env.current_date }}
-
name: Build and push icetray ml cuda 11.8 for 22 04
uses: docker/build-push-action@v3
with:
push: true
file: icecube_ml/Dockerfile_ubuntu_22.04_cuda11.8_cudann8
build-args: |
VERSION=${{ env.ICETRAY_VERSION }}
tags: >
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda11.8-cudnn8-ubuntu22.04,
icecube/icetray-ml:icetray-current-cuda11.8-cudnn8-ubuntu22.04,
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda11.8-cudnn8-ubuntu22.04-${{ env.current_date }}
-
name: Build and push icetray ml cuda 11.8 for 22 04 with graphnet
uses: docker/build-push-action@v3
with:
push: true
file: icecube_ml/Dockerfile_ubuntu_22.04_cuda11.8_cudann8_icetray_graphnet
build-args: |
VERSION=${{ env.ICETRAY_VERSION }}
tags: >
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda11.8-cudnn8-graphnet-ubuntu22.04,
icecube/icetray-ml:icetray-current-cuda11.8-cudnn8-graphnet-ubuntu22.04,
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda11.8-cudnn8-graphnet-ubuntu22.04-${{ env.current_date }}
-
name: Build and push icetray ml cuda 11.8 for 22 04 with tensorflow
uses: docker/build-push-action@v3
with:
push: true
file: icecube_ml/Dockerfile_ubuntu_22.04_cuda11.8_cudann8_icetray_tensorflow
build-args: |
VERSION=${{ env.ICETRAY_VERSION }}
tags: >
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda11.8-cudnn8-tf-ubuntu22.04,
icecube/icetray-ml:icetray-current-cuda11.8-cudnn8-tf-ubuntu22.04,
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda11.8-cudnn8-tf-ubuntu22.04-${{ env.current_date }}
-
name: Build and push icetray ml cuda 11.8 for 22 04 with graphnet dev
uses: docker/build-push-action@v3
with:
push: true
file: icecube_ml/Dockerfile_ubuntu_22.04_cuda11.8_cudann8_icetray_graphnet-devel
build-args: |
VERSION=${{ env.ICETRAY_VERSION }}
tags: >
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda11.8-cudnn8-graphnet-ubuntu22.04-devel,
icecube/icetray-ml:icetray-current-cuda11.8-cudnn8-graphnet-ubuntu22.04-devel,
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda11.8-cudnn8-graphnet-ubuntu22.04-devel-${{ env.current_date }}
-
name: Build and push icetray ml for 22 04 rocm
uses: docker/build-push-action@v3
with:
push: true
file: icecube_ml/Dockerfile_ubuntu_22.04_rocm5.7
build-args: |
VERSION=${{ env.ICETRAY_VERSION }}
tags: >
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-rocm5.7-ubuntu22.04-devel,
icecube/icetray-ml:icetray-current-rocm5.7-ubuntu22.04-devel,
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-rocm5.7-ubuntu22.04-devel-${{ env.current_date }}
-
name: Build and push icetray ml for 22 04 pytorch
uses: docker/build-push-action@v3
with:
push: true
file: icecube_ml/Dockerfile_ubuntu_22.04_cuda12.1_cudann_pytorch2.2.2
build-args: |
VERSION=${{ env.ICETRAY_VERSION }}
tags: >
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda12.1-cudann-pytorch2.2.2-ubuntu22.04-devel,
icecube/icetray-ml:icetray-current-cuda12.1-cudann-pytorch2.2.2-ubuntu22.04-devel,
icecube/icetray-ml:icetray-${{ env.ICETRAY_VERSION }}-cuda12.1-cudann-pytorch2.2.2-ubuntu22.04-devel-${{ env.current_date }}
# -
# name: Build and push icetray ml 20 04
# uses: docker/build-push-action@v3
# with:
# push: true
# file: icecube_ml/Dockerfile_ubuntu_20.04_icecube_ml
# build-args: |
# VERSION=${{ env.ICETRAY_VERSION }}
# tags: >
# icecube/icetray-ml:icetray-ml-base-${{ env.ICETRAY_VERSION }}-ubuntu20.04,
# icecube/icetray-ml:icetray-ml-base-current-ubuntu20.04,
# icecube/icetray-ml:icetray-ml-base-${{ env.ICETRAY_VERSION }}-ubuntu20.04-${{ env.current_date }}