-
Notifications
You must be signed in to change notification settings - Fork 83
83 lines (81 loc) · 2.5 KB
/
dockerhub.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
name: 'dockerhub'
on:
push:
branches:
- master
jobs:
build-bagua:
strategy:
fail-fast: false
matrix:
include:
- cuda-version: "cuda10.2"
cudnn-version: "cudnn7"
pytorch-version: "pytorch-1.9.0"
- cuda-version: "cuda11.1"
cudnn-version: "cudnn8"
pytorch-version: "pytorch-1.9.0"
- cuda-version: "cuda11.3"
cudnn-version: "cudnn8"
pytorch-version: "pytorch-1.10.0"
- cuda-version: "cuda11.3"
cudnn-version: "cudnn8"
pytorch-version: "pytorch-1.11.0"
- cuda-version: "cuda11.3"
cudnn-version: "cudnn8"
pytorch-version: "pytorch-1.12.0"
- cuda-version: "cuda11.6"
cudnn-version: "cudnn8"
pytorch-version: "pytorch-1.13.0"
- cuda-version: "cuda11.7"
cudnn-version: "cudnn8"
pytorch-version: "pytorch-2.0.0"
name: 'Build'
runs-on: ubuntu-latest
steps:
- name: "Build:checkout"
uses: actions/checkout@v2
with:
submodules: recursive
- name: Login to DockerHub
uses: docker/login-action@v1
with:
username: ${{ secrets.DOCKERHUB_USERNAME }}
password: ${{ secrets.DOCKERHUB_TOKEN }}
- name: build and push bagua
uses: docker/[email protected]
with:
context: .
push: true
tags: baguasys/bagua:master-${{ matrix.pytorch-version }}-${{ matrix.cuda-version }}-${{ matrix.cudnn-version }}
file: docker/Dockerfile.${{ matrix.pytorch-version }}-${{ matrix.cuda-version }}-${{ matrix.cudnn-version }}
build-manylinux:
strategy:
fail-fast: false
matrix:
cuda-version:
- "102"
- "111"
- "113"
- "115"
- "116"
- "117"
name: 'Build'
runs-on: ubuntu-latest
steps:
- name: "Build:checkout"
uses: actions/checkout@v2
with:
submodules: recursive
- name: Login to DockerHub
uses: docker/login-action@v1
with:
username: ${{ secrets.DOCKERHUB_USERNAME }}
password: ${{ secrets.DOCKERHUB_TOKEN }}
- name: build and push bagua/manylinux
uses: docker/[email protected]
with:
context: .
push: true
tags: baguasys/manylinux-cuda:${{ matrix.cuda-version }}
file: docker/Dockerfile.manylinux-cuda${{ matrix.cuda-version }}