config.yml 29.2 KB
Newer Older
1
2
3
4
version: 2.1

# How to test the Linux jobs:
#   - Install CircleCI local CLI: https://circleci.com/docs/2.0/local-cli/
Edward Z. Yang's avatar
Edward Z. Yang committed
5
6
7
#   - circleci config process .circleci/config.yml > gen.yml && circleci local execute -c gen.yml --job binary_linux_wheel_py3.7
#     - Replace binary_linux_wheel_py3.7 with the name of the job you want to test.
#       Job names are 'name:' key.
8

Francisco Massa's avatar
Francisco Massa committed
9
orbs:
10
  win: circleci/windows@2.0.0
Francisco Massa's avatar
Francisco Massa committed
11

12
13
14
15
16
17
18
executors:
  windows-gpu-prototype:
    machine:
      resource_class: windows.gpu.small.prototype
      image: windows-server-2019-nvidia:201908-28
      shell: bash.exe

19
20
21
22
23
commands:
  checkout_merge:
    description: "checkout merge branch"
    steps:
      - checkout
24
25
26
27
28
29
30
31
32
#     - run:
#         name: Checkout merge branch
#         command: |
#           set -ex
#           BRANCH=$(git rev-parse --abbrev-ref HEAD)
#           if [[ "$BRANCH" != "master" ]]; then
#             git fetch --force origin ${CIRCLE_BRANCH}/merge:merged/${CIRCLE_BRANCH}
#             git checkout "merged/$CIRCLE_BRANCH"
#           fi
33

34
35
36
37
38
39
40
41
binary_common: &binary_common
  parameters:
    # Edit these defaults to do a release`
    build_version:
      description: "version number of release binary; by default, build a nightly"
      type: string
      default: ""
    pytorch_version:
42
      description: "PyTorch version to build against; by default, use a nightly"
43
      type: string
44
      default: ""
45
46
47
48
    # Don't edit these
    python_version:
      description: "Python version to build against (e.g., 3.7)"
      type: string
Edward Z. Yang's avatar
Edward Z. Yang committed
49
50
    cu_version:
      description: "CUDA version to build against, in CU format (e.g., cpu or cu100)"
51
52
53
54
55
      type: string
    unicode_abi:
      description: "Python 2.7 wheel only: whether or not we are cp27mu (default: no)"
      type: string
      default: ""
Edward Z. Yang's avatar
Edward Z. Yang committed
56
57
58
    wheel_docker_image:
      description: "Wheel only: what docker image to use"
      type: string
59
      default: "pytorch/manylinux-cuda101"
60
61
62
63
64
  environment:
    PYTHON_VERSION: << parameters.python_version >>
    BUILD_VERSION: << parameters.build_version >>
    PYTORCH_VERSION: << parameters.pytorch_version >>
    UNICODE_ABI: << parameters.unicode_abi >>
Edward Z. Yang's avatar
Edward Z. Yang committed
65
    CU_VERSION: << parameters.cu_version >>
66

67
jobs:
68
69
70
71
72
73
74
  circleci_consistency:
    docker:
      - image: circleci/python:3.7
    steps:
      - checkout
      - run:
          command: |
75
            pip install --user --progress-bar off jinja2 pyyaml
76
77
78
79
80
81
            python .circleci/regenerate.py
            git diff --exit-code || (echo ".circleci/config.yml not in sync with config.yml.in! Run .circleci/regenerate.py to update config"; exit 1)

  binary_linux_wheel:
    <<: *binary_common
    docker:
Edward Z. Yang's avatar
Edward Z. Yang committed
82
      - image: << parameters.wheel_docker_image >>
83
84
    resource_class: 2xlarge+
    steps:
85
      - checkout_merge
86
87
88
      - run: packaging/build_wheel.sh
      - store_artifacts:
          path: dist
Edward Z. Yang's avatar
Edward Z. Yang committed
89
90
91
92
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
93
94
95
96

  binary_linux_conda:
    <<: *binary_common
    docker:
97
      - image: "pytorch/conda-cuda"
98
    resource_class: 2xlarge+
99
    steps:
100
      - checkout_merge
101
102
103
      - run: packaging/build_conda.sh
      - store_artifacts:
          path: /opt/conda/conda-bld/linux-64
Edward Z. Yang's avatar
Edward Z. Yang committed
104
105
106
107
      - persist_to_workspace:
          root: /opt/conda/conda-bld/linux-64
          paths:
            - "*"
108

Francisco Massa's avatar
Francisco Massa committed
109
110
111
112
113
114
  binary_linux_conda_cuda:
    <<: *binary_common
    machine:
      image: ubuntu-1604:201903-01
    resource_class: gpu.medium
    steps:
115
    - checkout_merge
Francisco Massa's avatar
Francisco Massa committed
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
    - run:
        name: Setup environment
        command: |
          set -e

          curl -L https://packagecloud.io/circleci/trusty/gpgkey | sudo apt-key add -
          curl -L https://dl.google.com/linux/linux_signing_key.pub | sudo apt-key add -

          sudo apt-get update

          sudo apt-get install \
              apt-transport-https \
              ca-certificates \
              curl \
              gnupg-agent \
              software-properties-common

          curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

          sudo add-apt-repository \
             "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
             $(lsb_release -cs) \
             stable"

          sudo apt-get update
          export DOCKER_VERSION="5:19.03.2~3-0~ubuntu-xenial"
Francisco Massa's avatar
Francisco Massa committed
142
          sudo apt-get install docker-ce=${DOCKER_VERSION} docker-ce-cli=${DOCKER_VERSION} containerd.io=1.2.6-3
Francisco Massa's avatar
Francisco Massa committed
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161

          # Add the package repositories
          distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
          curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
          curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

          export NVIDIA_CONTAINER_VERSION="1.0.3-1"
          sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit=${NVIDIA_CONTAINER_VERSION}
          sudo systemctl restart docker

          DRIVER_FN="NVIDIA-Linux-x86_64-410.104.run"
          wget "https://s3.amazonaws.com/ossci-linux/nvidia_driver/$DRIVER_FN"
          sudo /bin/bash "$DRIVER_FN" -s --no-drm || (sudo cat /var/log/nvidia-installer.log && false)
          nvidia-smi

    - run:
        name: Pull docker image
        command: |
          set -e
162
          export DOCKER_IMAGE=pytorch/conda-cuda
Francisco Massa's avatar
Francisco Massa committed
163
164
165
166
167
168
169
170
171
172
          echo Pulling docker image $DOCKER_IMAGE
          docker pull $DOCKER_IMAGE >/dev/null

    - run:
        name: Build and run tests
        command: |
          set -e

          cd ${HOME}/project/

173
          export DOCKER_IMAGE=pytorch/conda-cuda
Francisco Massa's avatar
Francisco Massa committed
174
175
176
177
178
179
180
          export VARS_TO_PASS="-e PYTHON_VERSION -e BUILD_VERSION -e PYTORCH_VERSION -e UNICODE_ABI -e CU_VERSION"

          docker run --gpus all  --ipc=host -v $(pwd):/remote -w /remote ${VARS_TO_PASS} ${DOCKER_IMAGE} ./packaging/build_conda.sh

  binary_win_conda:
    <<: *binary_common
    executor:
181
      name: win/default
Francisco Massa's avatar
Francisco Massa committed
182
183
      shell: bash.exe
    steps:
184
      - checkout_merge
Francisco Massa's avatar
Francisco Massa committed
185
186
187
      - run:
          command: |
            choco install miniconda3
188
189
190
            (& "C:\tools\miniconda3\Scripts\conda.exe" "shell.powershell" "hook") | Out-String | Invoke-Expression
            conda activate base
            conda install -yq conda-build "conda-package-handling!=1.5.0"
Francisco Massa's avatar
Francisco Massa committed
191
192
193
            bash packaging/build_conda.sh
          shell: powershell.exe

194
195
196
197
198
199
200
201
202
203
204
205
206
207
  binary_win_conda_cuda:
    <<: *binary_common
    executor: windows-gpu-prototype
    steps:
      - checkout_merge
      - run:
          command: |
            choco install miniconda3
            (& "C:\tools\miniconda3\Scripts\conda.exe" "shell.powershell" "hook") | Out-String | Invoke-Expression
            conda activate base
            conda install -yq conda-build "conda-package-handling!=1.5.0"
            bash packaging/build_conda.sh
          shell: powershell.exe

208
209
210
211
212
  binary_macos_wheel:
    <<: *binary_common
    macos:
      xcode: "9.0"
    steps:
213
      - checkout_merge
214
215
216
217
218
219
220
221
222
223
224
      - run:
          # Cannot easily deduplicate this as source'ing activate
          # will set environment variables which we need to propagate
          # to build_wheel.sh
          command: |
            curl -o conda.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh
            sh conda.sh -b
            source $HOME/miniconda3/bin/activate
            packaging/build_wheel.sh
      - store_artifacts:
          path: dist
Edward Z. Yang's avatar
Edward Z. Yang committed
225
226
227
228
      - persist_to_workspace:
          root: dist
          paths:
            - "*"
229
230
231
232
233
234

  binary_macos_conda:
    <<: *binary_common
    macos:
      xcode: "9.0"
    steps:
235
      - checkout_merge
236
237
238
239
240
241
242
243
244
      - run:
          command: |
            curl -o conda.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh
            sh conda.sh -b
            source $HOME/miniconda3/bin/activate
            conda install -yq conda-build
            packaging/build_conda.sh
      - store_artifacts:
          path: /Users/distiller/miniconda3/conda-bld/osx-64
Edward Z. Yang's avatar
Edward Z. Yang committed
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
      - persist_to_workspace:
          root: /Users/distiller/miniconda3/conda-bld/osx-64
          paths:
            - "*"

  # Requires org-member context
  binary_conda_upload:
    docker:
      - image: continuumio/miniconda
    steps:
      - attach_workspace:
          at: ~/workspace
      - run:
          command: |
            # Prevent credential from leaking
            conda install -yq anaconda-client
            set +x
            anaconda login \
                --username "$PYTORCH_BINARY_PJH5_CONDA_USERNAME" \
                --password "$PYTORCH_BINARY_PJH5_CONDA_PASSWORD"
            set -x
            anaconda upload ~/workspace/*.tar.bz2 -u pytorch-nightly --label main --no-progress --force

  # Requires org-member context
  binary_wheel_upload:
Edward Z. Yang's avatar
Edward Z. Yang committed
270
271
272
273
    parameters:
      subfolder:
        description: "What whl subfolder to upload to, e.g., blank or cu100/ (trailing slash is important)"
        type: string
Edward Z. Yang's avatar
Edward Z. Yang committed
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
    docker:
      - image: circleci/python:3.7
    steps:
      - attach_workspace:
          at: ~/workspace
      - checkout
      - run:
          command: |
            pip install --user awscli
            export PATH="$HOME/.local/bin:$PATH"
            # Prevent credential from leaking
            set +x
            export AWS_ACCESS_KEY_ID="${PYTORCH_BINARY_AWS_ACCESS_KEY_ID}"
            export AWS_SECRET_ACCESS_KEY="${PYTORCH_BINARY_AWS_SECRET_ACCESS_KEY}"
            set -x
Edward Z. Yang's avatar
Edward Z. Yang committed
289
290
291
            for pkg in ~/workspace/*.whl; do
              aws s3 cp "$pkg" "s3://pytorch/whl/nightly/<< parameters.subfolder >>" --acl public-read
            done
292

293

294
295
296
297
workflows:
  build:
    jobs:
      - circleci_consistency
Edward Z. Yang's avatar
Edward Z. Yang committed
298
      - binary_linux_wheel:
299
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
300
          name: binary_linux_wheel_py3.5_cpu
301
          python_version: '3.5'
Edward Z. Yang's avatar
Edward Z. Yang committed
302
      - binary_linux_wheel:
303
          cu_version: cu92
Edward Z. Yang's avatar
Edward Z. Yang committed
304
          name: binary_linux_wheel_py3.5_cu92
305
          python_version: '3.5'
306
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
307
      - binary_linux_wheel:
308
          cu_version: cu100
Edward Z. Yang's avatar
Edward Z. Yang committed
309
          name: binary_linux_wheel_py3.5_cu100
310
          python_version: '3.5'
311
          wheel_docker_image: pytorch/manylinux-cuda100
Francisco Massa's avatar
Francisco Massa committed
312
313
314
315
      - binary_linux_wheel:
          cu_version: cu101
          name: binary_linux_wheel_py3.5_cu101
          python_version: '3.5'
Edward Z. Yang's avatar
Edward Z. Yang committed
316
      - binary_linux_wheel:
317
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
318
          name: binary_linux_wheel_py3.6_cpu
319
          python_version: '3.6'
Edward Z. Yang's avatar
Edward Z. Yang committed
320
      - binary_linux_wheel:
321
          cu_version: cu92
Edward Z. Yang's avatar
Edward Z. Yang committed
322
          name: binary_linux_wheel_py3.6_cu92
323
          python_version: '3.6'
324
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
325
      - binary_linux_wheel:
326
          cu_version: cu100
Edward Z. Yang's avatar
Edward Z. Yang committed
327
          name: binary_linux_wheel_py3.6_cu100
328
          python_version: '3.6'
329
          wheel_docker_image: pytorch/manylinux-cuda100
Francisco Massa's avatar
Francisco Massa committed
330
331
332
333
      - binary_linux_wheel:
          cu_version: cu101
          name: binary_linux_wheel_py3.6_cu101
          python_version: '3.6'
Edward Z. Yang's avatar
Edward Z. Yang committed
334
      - binary_linux_wheel:
335
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
336
          name: binary_linux_wheel_py3.7_cpu
337
          python_version: '3.7'
Edward Z. Yang's avatar
Edward Z. Yang committed
338
      - binary_linux_wheel:
339
          cu_version: cu92
Edward Z. Yang's avatar
Edward Z. Yang committed
340
          name: binary_linux_wheel_py3.7_cu92
341
          python_version: '3.7'
342
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
343
      - binary_linux_wheel:
344
          cu_version: cu100
Edward Z. Yang's avatar
Edward Z. Yang committed
345
          name: binary_linux_wheel_py3.7_cu100
346
          python_version: '3.7'
347
          wheel_docker_image: pytorch/manylinux-cuda100
Francisco Massa's avatar
Francisco Massa committed
348
349
350
351
      - binary_linux_wheel:
          cu_version: cu101
          name: binary_linux_wheel_py3.7_cu101
          python_version: '3.7'
Edward Z. Yang's avatar
Edward Z. Yang committed
352
      - binary_macos_wheel:
353
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
354
          name: binary_macos_wheel_py3.5_cpu
355
          python_version: '3.5'
Edward Z. Yang's avatar
Edward Z. Yang committed
356
      - binary_macos_wheel:
357
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
358
          name: binary_macos_wheel_py3.6_cpu
359
          python_version: '3.6'
Edward Z. Yang's avatar
Edward Z. Yang committed
360
      - binary_macos_wheel:
361
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
362
          name: binary_macos_wheel_py3.7_cpu
363
          python_version: '3.7'
Edward Z. Yang's avatar
Edward Z. Yang committed
364
      - binary_linux_conda:
365
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
366
          name: binary_linux_conda_py3.5_cpu
367
          python_version: '3.5'
Edward Z. Yang's avatar
Edward Z. Yang committed
368
      - binary_linux_conda:
369
          cu_version: cu92
Edward Z. Yang's avatar
Edward Z. Yang committed
370
          name: binary_linux_conda_py3.5_cu92
371
          python_version: '3.5'
372
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
373
      - binary_linux_conda:
374
          cu_version: cu100
Edward Z. Yang's avatar
Edward Z. Yang committed
375
          name: binary_linux_conda_py3.5_cu100
376
          python_version: '3.5'
377
          wheel_docker_image: pytorch/manylinux-cuda100
Francisco Massa's avatar
Francisco Massa committed
378
379
380
381
      - binary_linux_conda:
          cu_version: cu101
          name: binary_linux_conda_py3.5_cu101
          python_version: '3.5'
Edward Z. Yang's avatar
Edward Z. Yang committed
382
      - binary_linux_conda:
383
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
384
          name: binary_linux_conda_py3.6_cpu
385
          python_version: '3.6'
Edward Z. Yang's avatar
Edward Z. Yang committed
386
      - binary_linux_conda:
387
          cu_version: cu92
Edward Z. Yang's avatar
Edward Z. Yang committed
388
          name: binary_linux_conda_py3.6_cu92
389
          python_version: '3.6'
390
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
391
      - binary_linux_conda:
392
          cu_version: cu100
Edward Z. Yang's avatar
Edward Z. Yang committed
393
          name: binary_linux_conda_py3.6_cu100
394
          python_version: '3.6'
395
          wheel_docker_image: pytorch/manylinux-cuda100
Francisco Massa's avatar
Francisco Massa committed
396
397
398
399
      - binary_linux_conda:
          cu_version: cu101
          name: binary_linux_conda_py3.6_cu101
          python_version: '3.6'
Edward Z. Yang's avatar
Edward Z. Yang committed
400
      - binary_linux_conda:
401
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
402
          name: binary_linux_conda_py3.7_cpu
403
          python_version: '3.7'
Edward Z. Yang's avatar
Edward Z. Yang committed
404
      - binary_linux_conda:
405
          cu_version: cu92
Edward Z. Yang's avatar
Edward Z. Yang committed
406
          name: binary_linux_conda_py3.7_cu92
407
          python_version: '3.7'
408
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
409
      - binary_linux_conda:
410
          cu_version: cu100
Edward Z. Yang's avatar
Edward Z. Yang committed
411
          name: binary_linux_conda_py3.7_cu100
412
          python_version: '3.7'
413
          wheel_docker_image: pytorch/manylinux-cuda100
Francisco Massa's avatar
Francisco Massa committed
414
415
416
417
      - binary_linux_conda:
          cu_version: cu101
          name: binary_linux_conda_py3.7_cu101
          python_version: '3.7'
Edward Z. Yang's avatar
Edward Z. Yang committed
418
      - binary_macos_conda:
419
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
420
          name: binary_macos_conda_py3.5_cpu
421
          python_version: '3.5'
Edward Z. Yang's avatar
Edward Z. Yang committed
422
      - binary_macos_conda:
423
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
424
          name: binary_macos_conda_py3.6_cpu
425
          python_version: '3.6'
Edward Z. Yang's avatar
Edward Z. Yang committed
426
      - binary_macos_conda:
427
          cu_version: cpu
Edward Z. Yang's avatar
Edward Z. Yang committed
428
          name: binary_macos_conda_py3.7_cpu
429
          python_version: '3.7'
Francisco Massa's avatar
Francisco Massa committed
430
431
432
433
434
435
436
437
      - binary_linux_conda_cuda:
          name: torchvision_linux_py3.7_cu100
          python_version: "3.7"
          cu_version: "cu100"
      - binary_win_conda:
          name: torchvision_win_py3.6_cpu
          python_version: "3.6"
          cu_version: "cpu"
438
439
440
441
      - binary_win_conda_cuda:
          name: torchvision_win_py3.6_cu101
          python_version: "3.6"
          cu_version: "cu101"
Edward Z. Yang's avatar
Edward Z. Yang committed
442
443
444
445

  nightly:
    jobs:
      - circleci_consistency
446
      - binary_linux_wheel:
447
          cu_version: cpu
448
449
450
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
451
          name: nightly_binary_linux_wheel_py3.5_cpu
452
          python_version: '3.5'
Edward Z. Yang's avatar
Edward Z. Yang committed
453
454
      - binary_wheel_upload:
          context: org-member
455
456
457
          filters:
            branches:
              only: nightly
458
          name: nightly_binary_linux_wheel_py3.5_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
459
          requires:
460
461
          - nightly_binary_linux_wheel_py3.5_cpu
          subfolder: cpu/
462
      - binary_linux_wheel:
463
          cu_version: cu92
464
465
466
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
467
          name: nightly_binary_linux_wheel_py3.5_cu92
468
          python_version: '3.5'
469
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
470
471
      - binary_wheel_upload:
          context: org-member
472
473
474
          filters:
            branches:
              only: nightly
475
          name: nightly_binary_linux_wheel_py3.5_cu92_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
476
          requires:
477
478
          - nightly_binary_linux_wheel_py3.5_cu92
          subfolder: cu92/
479
      - binary_linux_wheel:
480
          cu_version: cu100
481
482
483
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
484
          name: nightly_binary_linux_wheel_py3.5_cu100
485
          python_version: '3.5'
486
          wheel_docker_image: pytorch/manylinux-cuda100
Edward Z. Yang's avatar
Edward Z. Yang committed
487
488
      - binary_wheel_upload:
          context: org-member
489
490
491
          filters:
            branches:
              only: nightly
492
          name: nightly_binary_linux_wheel_py3.5_cu100_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
493
          requires:
494
495
          - nightly_binary_linux_wheel_py3.5_cu100
          subfolder: cu100/
Francisco Massa's avatar
Francisco Massa committed
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
      - binary_linux_wheel:
          cu_version: cu101
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_wheel_py3.5_cu101
          python_version: '3.5'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_wheel_py3.5_cu101_upload
          requires:
          - nightly_binary_linux_wheel_py3.5_cu101
          subfolder: cu101/
512
      - binary_linux_wheel:
513
          cu_version: cpu
514
515
516
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
517
          name: nightly_binary_linux_wheel_py3.6_cpu
518
          python_version: '3.6'
Edward Z. Yang's avatar
Edward Z. Yang committed
519
520
      - binary_wheel_upload:
          context: org-member
521
522
523
          filters:
            branches:
              only: nightly
524
          name: nightly_binary_linux_wheel_py3.6_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
525
          requires:
526
527
          - nightly_binary_linux_wheel_py3.6_cpu
          subfolder: cpu/
528
      - binary_linux_wheel:
529
          cu_version: cu92
530
531
532
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
533
          name: nightly_binary_linux_wheel_py3.6_cu92
534
          python_version: '3.6'
535
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
536
537
      - binary_wheel_upload:
          context: org-member
538
539
540
          filters:
            branches:
              only: nightly
541
          name: nightly_binary_linux_wheel_py3.6_cu92_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
542
          requires:
543
544
          - nightly_binary_linux_wheel_py3.6_cu92
          subfolder: cu92/
545
      - binary_linux_wheel:
546
          cu_version: cu100
547
548
549
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
550
          name: nightly_binary_linux_wheel_py3.6_cu100
551
          python_version: '3.6'
552
          wheel_docker_image: pytorch/manylinux-cuda100
Edward Z. Yang's avatar
Edward Z. Yang committed
553
554
      - binary_wheel_upload:
          context: org-member
555
556
557
          filters:
            branches:
              only: nightly
558
          name: nightly_binary_linux_wheel_py3.6_cu100_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
559
          requires:
560
561
          - nightly_binary_linux_wheel_py3.6_cu100
          subfolder: cu100/
Francisco Massa's avatar
Francisco Massa committed
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
      - binary_linux_wheel:
          cu_version: cu101
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_wheel_py3.6_cu101
          python_version: '3.6'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_wheel_py3.6_cu101_upload
          requires:
          - nightly_binary_linux_wheel_py3.6_cu101
          subfolder: cu101/
578
      - binary_linux_wheel:
579
          cu_version: cpu
580
581
582
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
583
          name: nightly_binary_linux_wheel_py3.7_cpu
584
          python_version: '3.7'
Edward Z. Yang's avatar
Edward Z. Yang committed
585
586
      - binary_wheel_upload:
          context: org-member
587
588
589
          filters:
            branches:
              only: nightly
590
          name: nightly_binary_linux_wheel_py3.7_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
591
          requires:
592
593
          - nightly_binary_linux_wheel_py3.7_cpu
          subfolder: cpu/
594
      - binary_linux_wheel:
595
          cu_version: cu92
596
597
598
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
599
          name: nightly_binary_linux_wheel_py3.7_cu92
600
          python_version: '3.7'
601
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
602
603
      - binary_wheel_upload:
          context: org-member
604
605
606
          filters:
            branches:
              only: nightly
607
          name: nightly_binary_linux_wheel_py3.7_cu92_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
608
          requires:
609
610
          - nightly_binary_linux_wheel_py3.7_cu92
          subfolder: cu92/
611
      - binary_linux_wheel:
612
          cu_version: cu100
613
614
615
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
616
          name: nightly_binary_linux_wheel_py3.7_cu100
617
          python_version: '3.7'
618
          wheel_docker_image: pytorch/manylinux-cuda100
Edward Z. Yang's avatar
Edward Z. Yang committed
619
620
      - binary_wheel_upload:
          context: org-member
621
622
623
          filters:
            branches:
              only: nightly
624
          name: nightly_binary_linux_wheel_py3.7_cu100_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
625
          requires:
626
627
          - nightly_binary_linux_wheel_py3.7_cu100
          subfolder: cu100/
Francisco Massa's avatar
Francisco Massa committed
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
      - binary_linux_wheel:
          cu_version: cu101
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_wheel_py3.7_cu101
          python_version: '3.7'
      - binary_wheel_upload:
          context: org-member
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_wheel_py3.7_cu101_upload
          requires:
          - nightly_binary_linux_wheel_py3.7_cu101
          subfolder: cu101/
644
      - binary_macos_wheel:
645
          cu_version: cpu
646
647
648
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
649
          name: nightly_binary_macos_wheel_py3.5_cpu
650
          python_version: '3.5'
Edward Z. Yang's avatar
Edward Z. Yang committed
651
652
      - binary_wheel_upload:
          context: org-member
653
654
655
          filters:
            branches:
              only: nightly
656
          name: nightly_binary_macos_wheel_py3.5_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
657
          requires:
658
659
          - nightly_binary_macos_wheel_py3.5_cpu
          subfolder: ''
660
      - binary_macos_wheel:
661
          cu_version: cpu
662
663
664
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
665
          name: nightly_binary_macos_wheel_py3.6_cpu
666
          python_version: '3.6'
Edward Z. Yang's avatar
Edward Z. Yang committed
667
668
      - binary_wheel_upload:
          context: org-member
669
670
671
          filters:
            branches:
              only: nightly
672
          name: nightly_binary_macos_wheel_py3.6_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
673
          requires:
674
675
          - nightly_binary_macos_wheel_py3.6_cpu
          subfolder: ''
676
      - binary_macos_wheel:
677
          cu_version: cpu
678
679
680
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
681
          name: nightly_binary_macos_wheel_py3.7_cpu
682
          python_version: '3.7'
Edward Z. Yang's avatar
Edward Z. Yang committed
683
684
      - binary_wheel_upload:
          context: org-member
685
686
687
          filters:
            branches:
              only: nightly
688
          name: nightly_binary_macos_wheel_py3.7_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
689
          requires:
690
691
          - nightly_binary_macos_wheel_py3.7_cpu
          subfolder: ''
692
      - binary_linux_conda:
693
          cu_version: cpu
694
695
696
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
697
          name: nightly_binary_linux_conda_py3.5_cpu
698
          python_version: '3.5'
Edward Z. Yang's avatar
Edward Z. Yang committed
699
700
      - binary_conda_upload:
          context: org-member
701
702
703
          filters:
            branches:
              only: nightly
704
          name: nightly_binary_linux_conda_py3.5_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
705
          requires:
706
          - nightly_binary_linux_conda_py3.5_cpu
707
      - binary_linux_conda:
708
          cu_version: cu92
709
710
711
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
712
          name: nightly_binary_linux_conda_py3.5_cu92
713
          python_version: '3.5'
714
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
715
716
      - binary_conda_upload:
          context: org-member
717
718
719
          filters:
            branches:
              only: nightly
720
          name: nightly_binary_linux_conda_py3.5_cu92_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
721
          requires:
722
          - nightly_binary_linux_conda_py3.5_cu92
723
      - binary_linux_conda:
724
          cu_version: cu100
725
726
727
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
728
          name: nightly_binary_linux_conda_py3.5_cu100
729
          python_version: '3.5'
730
          wheel_docker_image: pytorch/manylinux-cuda100
Edward Z. Yang's avatar
Edward Z. Yang committed
731
732
      - binary_conda_upload:
          context: org-member
733
734
735
          filters:
            branches:
              only: nightly
736
          name: nightly_binary_linux_conda_py3.5_cu100_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
737
          requires:
738
          - nightly_binary_linux_conda_py3.5_cu100
Francisco Massa's avatar
Francisco Massa committed
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
      - binary_linux_conda:
          cu_version: cu101
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_conda_py3.5_cu101
          python_version: '3.5'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_conda_py3.5_cu101_upload
          requires:
          - nightly_binary_linux_conda_py3.5_cu101
754
      - binary_linux_conda:
755
          cu_version: cpu
756
757
758
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
759
          name: nightly_binary_linux_conda_py3.6_cpu
760
          python_version: '3.6'
Edward Z. Yang's avatar
Edward Z. Yang committed
761
762
      - binary_conda_upload:
          context: org-member
763
764
765
          filters:
            branches:
              only: nightly
766
          name: nightly_binary_linux_conda_py3.6_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
767
          requires:
768
          - nightly_binary_linux_conda_py3.6_cpu
769
      - binary_linux_conda:
770
          cu_version: cu92
771
772
773
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
774
          name: nightly_binary_linux_conda_py3.6_cu92
775
          python_version: '3.6'
776
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
777
778
      - binary_conda_upload:
          context: org-member
779
780
781
          filters:
            branches:
              only: nightly
782
          name: nightly_binary_linux_conda_py3.6_cu92_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
783
          requires:
784
          - nightly_binary_linux_conda_py3.6_cu92
785
      - binary_linux_conda:
786
          cu_version: cu100
787
788
789
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
790
          name: nightly_binary_linux_conda_py3.6_cu100
791
          python_version: '3.6'
792
          wheel_docker_image: pytorch/manylinux-cuda100
Edward Z. Yang's avatar
Edward Z. Yang committed
793
794
      - binary_conda_upload:
          context: org-member
795
796
797
          filters:
            branches:
              only: nightly
798
          name: nightly_binary_linux_conda_py3.6_cu100_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
799
          requires:
800
          - nightly_binary_linux_conda_py3.6_cu100
Francisco Massa's avatar
Francisco Massa committed
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
      - binary_linux_conda:
          cu_version: cu101
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_conda_py3.6_cu101
          python_version: '3.6'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_conda_py3.6_cu101_upload
          requires:
          - nightly_binary_linux_conda_py3.6_cu101
816
      - binary_linux_conda:
817
          cu_version: cpu
818
819
820
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
821
          name: nightly_binary_linux_conda_py3.7_cpu
822
          python_version: '3.7'
Edward Z. Yang's avatar
Edward Z. Yang committed
823
824
      - binary_conda_upload:
          context: org-member
825
826
827
          filters:
            branches:
              only: nightly
828
          name: nightly_binary_linux_conda_py3.7_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
829
          requires:
830
          - nightly_binary_linux_conda_py3.7_cpu
831
      - binary_linux_conda:
832
          cu_version: cu92
833
834
835
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
836
          name: nightly_binary_linux_conda_py3.7_cu92
837
          python_version: '3.7'
838
          wheel_docker_image: pytorch/manylinux-cuda92
Edward Z. Yang's avatar
Edward Z. Yang committed
839
840
      - binary_conda_upload:
          context: org-member
841
842
843
          filters:
            branches:
              only: nightly
844
          name: nightly_binary_linux_conda_py3.7_cu92_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
845
          requires:
846
          - nightly_binary_linux_conda_py3.7_cu92
847
      - binary_linux_conda:
848
          cu_version: cu100
849
850
851
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
852
          name: nightly_binary_linux_conda_py3.7_cu100
853
          python_version: '3.7'
854
          wheel_docker_image: pytorch/manylinux-cuda100
Edward Z. Yang's avatar
Edward Z. Yang committed
855
856
      - binary_conda_upload:
          context: org-member
857
858
859
          filters:
            branches:
              only: nightly
860
          name: nightly_binary_linux_conda_py3.7_cu100_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
861
          requires:
862
          - nightly_binary_linux_conda_py3.7_cu100
Francisco Massa's avatar
Francisco Massa committed
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
      - binary_linux_conda:
          cu_version: cu101
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_conda_py3.7_cu101
          python_version: '3.7'
      - binary_conda_upload:
          context: org-member
          filters:
            branches:
              only: nightly
          name: nightly_binary_linux_conda_py3.7_cu101_upload
          requires:
          - nightly_binary_linux_conda_py3.7_cu101
878
      - binary_macos_conda:
879
          cu_version: cpu
880
881
882
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
883
          name: nightly_binary_macos_conda_py3.5_cpu
884
          python_version: '3.5'
Edward Z. Yang's avatar
Edward Z. Yang committed
885
886
      - binary_conda_upload:
          context: org-member
887
888
889
          filters:
            branches:
              only: nightly
890
          name: nightly_binary_macos_conda_py3.5_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
891
          requires:
892
          - nightly_binary_macos_conda_py3.5_cpu
893
      - binary_macos_conda:
894
          cu_version: cpu
895
896
897
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
898
          name: nightly_binary_macos_conda_py3.6_cpu
899
          python_version: '3.6'
Edward Z. Yang's avatar
Edward Z. Yang committed
900
901
      - binary_conda_upload:
          context: org-member
902
903
904
          filters:
            branches:
              only: nightly
905
          name: nightly_binary_macos_conda_py3.6_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
906
          requires:
907
          - nightly_binary_macos_conda_py3.6_cpu
908
      - binary_macos_conda:
909
          cu_version: cpu
910
911
912
          filters:
            branches:
              only: nightly
Edward Z. Yang's avatar
Edward Z. Yang committed
913
          name: nightly_binary_macos_conda_py3.7_cpu
914
          python_version: '3.7'
Edward Z. Yang's avatar
Edward Z. Yang committed
915
916
      - binary_conda_upload:
          context: org-member
917
918
919
          filters:
            branches:
              only: nightly
920
          name: nightly_binary_macos_conda_py3.7_cpu_upload
Edward Z. Yang's avatar
Edward Z. Yang committed
921
          requires:
922
          - nightly_binary_macos_conda_py3.7_cpu