-
Notifications
You must be signed in to change notification settings - Fork 10
/
Copy path.gitlab-ci.yml
118 lines (111 loc) · 4 KB
/
.gitlab-ci.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
variables:
# This fixes issues with conda-build and versions of git < 1.9. Cloning from a
# shallow clone is not possible.
GIT_DEPTH: 0
GIT_STRATEGY: clone
stages:
- build
- publish
build-9.2:
stage: build
image: pytorch/manylinux-cuda92
variables:
cudatoolkit: "9.2"
# See:
# pytorch commit cd207737017db8c81584763207df20bc6110ed75
# https://en.wikipedia.org/wiki/CUDA#GPUs_supported
# https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#options-for-steering-gpu-code-generation-gpu-architecture
TORCH_CUDA_ARCH_LIST: "3.5 5.2 6.0 6.1"
script:
- wget -nv https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh -O install_conda.sh
- bash install_conda.sh -b
- export PATH="$HOME/miniconda3/bin:$PATH"
- mkdir -p /cache/pkgs
- rsync -a $HOME/miniconda3/pkgs /cache/pkgs
- rm -rf $HOME/miniconda3/pkgs
- ln -s /cache/pkgs $HOME/miniconda3/
- conda install -y -c pytorch pytorch=1.4 python=3.7
- conda install -y conda-build conda-verify
- conda build conda/ -c aahendriksen -c pytorch -c defaults -c conda-forge -m conda/cuda-${cudatoolkit}.yaml
- mkdir -p artifacts
- mv /root/miniconda3/conda-bld/linux-64/msd_pytorch*.bz2 artifacts/
artifacts:
paths:
- artifacts/
expire_in: 7 days
build-10.0:
stage: build
image: pytorch/manylinux-cuda100
variables:
cudatoolkit: "10.0"
# See:
# pytorch commit cd207737017db8c81584763207df20bc6110ed75
# https://en.wikipedia.org/wiki/CUDA#GPUs_supported
# https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#options-for-steering-gpu-code-generation-gpu-architecture
TORCH_CUDA_ARCH_LIST: "3.5 5.2 6.0 6.1 7.0+PTX"
script:
- wget -nv https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh -O install_conda.sh
- bash install_conda.sh -b
- export PATH="$HOME/miniconda3/bin:$PATH"
- conda install -y -c pytorch pytorch=1.4 python=3.7
- conda install -y conda-build conda-verify
- conda build conda/ -c aahendriksen -c pytorch -c defaults -c conda-forge -m conda/cuda-${cudatoolkit}.yaml
- mkdir -p artifacts
- mv /root/miniconda3/conda-bld/linux-64/msd_pytorch*.bz2 artifacts/
artifacts:
paths:
- artifacts/
expire_in: 7 days
build-10.1:
stage: build
image: pytorch/manylinux-cuda101
variables:
cudatoolkit: "10.1"
# See:
# pytorch commit cd207737017db8c81584763207df20bc6110ed75
# https://en.wikipedia.org/wiki/CUDA#GPUs_supported
# https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#options-for-steering-gpu-code-generation-gpu-architecture
TORCH_CUDA_ARCH_LIST: "3.5 5.2 6.0 6.1 7.0+PTX"
script:
- wget -nv https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh -O install_conda.sh
- bash install_conda.sh -b
- export PATH="$HOME/miniconda3/bin:$PATH"
- conda install -y -c pytorch pytorch=1.4 python=3.7
- conda install -y conda-build conda-verify
- conda build conda/ -c aahendriksen -c pytorch -c defaults -c conda-forge -m conda/cuda-${cudatoolkit}.yaml
- mkdir -p artifacts
- mv /root/miniconda3/conda-bld/linux-64/msd_pytorch*.bz2 artifacts/
artifacts:
paths:
- artifacts/
expire_in: 7 days
publish-dev:
stage: publish
image: continuumio/miniconda3
only:
- dev
variables:
ANACONDA_USERNAME: $ANACONDA_USERNAME
ANACONDA_PASSWORD: $ANACONDA_PASSWORD
script:
- conda install -yq anaconda-client
- set +x
- anaconda login --username "$ANACONDA_USERNAME" --password "$ANACONDA_PASSWORD"
- set -x
- anaconda upload --label dev artifacts/*.bz2
- anaconda logout
publish-release:
stage: publish
image: continuumio/miniconda3
only:
- tags
variables:
ANACONDA_USERNAME: $ANACONDA_USERNAME
ANACONDA_PASSWORD: $ANACONDA_PASSWORD
script:
- conda install -yq anaconda-client
- set +x
- anaconda login --username "$ANACONDA_USERNAME" --password "$ANACONDA_PASSWORD"
- set -x
- anaconda upload artifacts/*.bz2
- anaconda logout