config.yml 24.4 KB
Newer Older
facebook-github-bot's avatar
facebook-github-bot committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
version: 2.1

#examples:
#https://github.com/facebookresearch/ParlAI/blob/master/.circleci/config.yml
#https://github.com/facebookresearch/hydra/blob/master/.circleci/config.yml
#https://github.com/facebookresearch/habitat-api/blob/master/.circleci/config.yml

#drive tests with nox or tox or pytest?

# -------------------------------------------------------------------------------------
# environments where we run our jobs
# -------------------------------------------------------------------------------------


setupcuda: &setupcuda
  run:
    name: Setup CUDA
    working_directory: ~/
    command: |
      # download and install nvidia drivers, cuda, etc
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
21
22
      wget --no-verbose --no-clobber -P ~/nvidia-downloads https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux.run
      sudo sh ~/nvidia-downloads/cuda_11.2.2_460.32.03_linux.run --silent
facebook-github-bot's avatar
facebook-github-bot committed
23
24
25
      echo "Done installing CUDA."
      pyenv versions
      nvidia-smi
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
26
      pyenv global 3.9.1
facebook-github-bot's avatar
facebook-github-bot committed
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

gpu: &gpu
  environment:
    CUDA_VERSION: "10.2"
  machine:
    image: default
  resource_class: gpu.medium # tesla m60

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:
      description: "PyTorch version to build against; by default, use a nightly"
      type: string
      default: ""
    # Don't edit these
    python_version:
      description: "Python version to build against (e.g., 3.7)"
      type: string
    cu_version:
      description: "CUDA version to build against, in CU format (e.g., cpu or cu100)"
      type: string
    wheel_docker_image:
      description: "Wheel only: what docker image to use"
      type: string
56
      default: "pytorch/manylinux-cuda101"
facebook-github-bot's avatar
facebook-github-bot committed
57
58
59
60
61
62
63
64
65
66
  environment:
    PYTHON_VERSION: << parameters.python_version >>
    BUILD_VERSION: << parameters.build_version >>
    PYTORCH_VERSION: << parameters.pytorch_version >>
    CU_VERSION: << parameters.cu_version >>

jobs:
  main:
    <<: *gpu
    machine:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
67
      image: ubuntu-2004:202101-01
facebook-github-bot's avatar
facebook-github-bot committed
68
69
70
    steps:
      - checkout
      - <<: *setupcuda
71
      - run: pip3 install --progress-bar off imageio wheel matplotlib 'pillow<7'
facebook-github-bot's avatar
facebook-github-bot committed
72
73
74
75
76
77
      - run: pip3 install --progress-bar off torch torchvision
      # - run: conda create -p ~/conda_env python=3.7 numpy
      # - run: conda activate ~/conda_env
      # - run: conda install -c pytorch pytorch torchvision

      - run: pip3 install --progress-bar off 'git+https://github.com/facebookresearch/fvcore'
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
78
      - run: pip3 install --progress-bar off 'git+https://github.com/facebookresearch/iopath'
79
80
81
82
83
84
85
86
87
88
      - run:
          name: get cub
          command: |
            cd ..
            wget --no-verbose https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz
            tar xzf 1.10.0.tar.gz
            # This expands to a directory called cub-1.10.0
      - run:
          name: build
          command: |
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
89
            export LD_LIBRARY_PATH=$LD_LIBARY_PATH:/usr/local/cuda-11.2/lib64
90
91
            export CUB_HOME=$(realpath ../cub-1.10.0)
            python3 setup.py build_ext --inplace
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
92
      - run: LD_LIBRARY_PATH=$LD_LIBARY_PATH:/usr/local/cuda-11.2/lib64 python -m unittest discover -v -s tests
facebook-github-bot's avatar
facebook-github-bot committed
93
94
95
96
97
98
      - run: python3 setup.py bdist_wheel

  binary_linux_wheel:
    <<: *binary_common
    docker:
      - image: << parameters.wheel_docker_image >>
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
99
100
101
        auth:
          username: $DOCKERHUB_USERNAME
          password: $DOCKERHUB_TOKEN
facebook-github-bot's avatar
facebook-github-bot committed
102
103
104
    resource_class: 2xlarge+
    steps:
      - checkout
105
      - run: MAX_JOBS=15 packaging/build_wheel.sh
facebook-github-bot's avatar
facebook-github-bot committed
106
107
108
109
110
111
112
113
114
115
116
      - store_artifacts:
          path: dist
      - persist_to_workspace:
          root: dist
          paths:
            - "*"

  binary_linux_conda:
    <<: *binary_common
    docker:
      - image: "pytorch/conda-cuda"
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
117
118
119
        auth:
          username: $DOCKERHUB_USERNAME
          password: $DOCKERHUB_TOKEN
facebook-github-bot's avatar
facebook-github-bot committed
120
121
122
123
124
    resource_class: 2xlarge+
    steps:
      - checkout
      # This is building with cuda but no gpu present,
      # so we aren't running the tests.
125
      - run: MAX_JOBS=15 TEST_FLAG=--no-test packaging/build_conda.sh
facebook-github-bot's avatar
facebook-github-bot committed
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
      - store_artifacts:
          path: /opt/conda/conda-bld/linux-64
      - persist_to_workspace:
          root: /opt/conda/conda-bld/linux-64
          paths:
            - "*"

  binary_linux_conda_cuda:
    <<: *binary_common
    machine:
      image: ubuntu-1604:201903-01
    resource_class: gpu.medium
    steps:
    - checkout
    - 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"
          sudo apt-get install docker-ce=${DOCKER_VERSION} docker-ce-cli=${DOCKER_VERSION} containerd.io=1.2.6-3

          # 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

Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
177
178
          DRIVER_FN="NVIDIA-Linux-x86_64-450.80.02.run"
          wget "https://us.download.nvidia.com/XFree86/Linux-x86_64/450.80.02/$DRIVER_FN"
facebook-github-bot's avatar
facebook-github-bot committed
179
180
181
182
183
184
185
          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
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
186
187
188

          { docker login -u="$DOCKERHUB_USERNAME" -p="$DOCKERHUB_TOKEN" ; } 2> /dev/null

Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
189
          DOCKER_IMAGE=pytorch/conda-cuda
facebook-github-bot's avatar
facebook-github-bot committed
190
          echo Pulling docker image $DOCKER_IMAGE
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
191
          docker pull $DOCKER_IMAGE
facebook-github-bot's avatar
facebook-github-bot committed
192
193
    - run:
        name: Build and run tests
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
194
        no_output_timeout: 20m
facebook-github-bot's avatar
facebook-github-bot committed
195
196
197
198
199
        command: |
          set -e

          cd ${HOME}/project/

Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
200
201
202
          DOCKER_IMAGE=pytorch/conda-cuda
          export JUST_TESTRUN=1
          VARS_TO_PASS="-e PYTHON_VERSION -e BUILD_VERSION -e PYTORCH_VERSION -e CU_VERSION -e JUST_TESTRUN"
facebook-github-bot's avatar
facebook-github-bot committed
203
204
205

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

206
207
208
  binary_macos_wheel:
    <<: *binary_common
    macos:
209
      xcode: "9.4.1"
210
211
212
213
214
215
216
217
218
219
220
221
222
223
    steps:
      - checkout
      - 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

facebook-github-bot's avatar
facebook-github-bot committed
224
225
226
227
workflows:
  version: 2
  build_and_test:
    jobs:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
228
229
      - main:
          context: DOCKERHUB_TOKEN
facebook-github-bot's avatar
facebook-github-bot committed
230
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
231
          context: DOCKERHUB_TOKEN
facebook-github-bot's avatar
facebook-github-bot committed
232
          cu_version: cu92
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
233
          name: linux_conda_py36_cu92_pyt14
facebook-github-bot's avatar
facebook-github-bot committed
234
235
236
          python_version: '3.6'
          pytorch_version: '1.4'
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
237
          context: DOCKERHUB_TOKEN
facebook-github-bot's avatar
facebook-github-bot committed
238
          cu_version: cu101
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
239
          name: linux_conda_py36_cu101_pyt14
facebook-github-bot's avatar
facebook-github-bot committed
240
241
242
          python_version: '3.6'
          pytorch_version: '1.4'
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
243
          context: DOCKERHUB_TOKEN
facebook-github-bot's avatar
facebook-github-bot committed
244
          cu_version: cu92
245
          name: linux_conda_py36_cu92_pyt150
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
246
          python_version: '3.6'
247
          pytorch_version: 1.5.0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
248
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
249
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
250
          cu_version: cu101
251
          name: linux_conda_py36_cu101_pyt150
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
252
          python_version: '3.6'
253
          pytorch_version: 1.5.0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
254
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
255
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
256
          cu_version: cu102
257
          name: linux_conda_py36_cu102_pyt150
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
258
          python_version: '3.6'
259
260
          pytorch_version: 1.5.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
261
          context: DOCKERHUB_TOKEN
262
263
264
265
266
          cu_version: cu92
          name: linux_conda_py36_cu92_pyt151
          python_version: '3.6'
          pytorch_version: 1.5.1
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
267
          context: DOCKERHUB_TOKEN
268
269
270
271
272
          cu_version: cu101
          name: linux_conda_py36_cu101_pyt151
          python_version: '3.6'
          pytorch_version: 1.5.1
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
273
          context: DOCKERHUB_TOKEN
274
275
276
277
          cu_version: cu102
          name: linux_conda_py36_cu102_pyt151
          python_version: '3.6'
          pytorch_version: 1.5.1
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
278
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
279
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
280
281
282
283
284
          cu_version: cu92
          name: linux_conda_py36_cu92_pyt160
          python_version: '3.6'
          pytorch_version: 1.6.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
285
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
286
287
288
289
290
          cu_version: cu101
          name: linux_conda_py36_cu101_pyt160
          python_version: '3.6'
          pytorch_version: 1.6.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
291
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
292
293
294
295
          cu_version: cu102
          name: linux_conda_py36_cu102_pyt160
          python_version: '3.6'
          pytorch_version: 1.6.0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
296
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
297
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
298
299
300
301
302
          cu_version: cu101
          name: linux_conda_py36_cu101_pyt170
          python_version: '3.6'
          pytorch_version: 1.7.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
303
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
304
305
306
307
308
          cu_version: cu102
          name: linux_conda_py36_cu102_pyt170
          python_version: '3.6'
          pytorch_version: 1.7.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
309
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
310
311
312
313
          cu_version: cu110
          name: linux_conda_py36_cu110_pyt170
          python_version: '3.6'
          pytorch_version: 1.7.0
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py36_cu101_pyt171
          python_version: '3.6'
          pytorch_version: 1.7.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py36_cu102_pyt171
          python_version: '3.6'
          pytorch_version: 1.7.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu110
          name: linux_conda_py36_cu110_pyt171
          python_version: '3.6'
          pytorch_version: 1.7.1
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py36_cu101_pyt180
          python_version: '3.6'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py36_cu102_pyt180
          python_version: '3.6'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu111
          name: linux_conda_py36_cu111_pyt180
          python_version: '3.6'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py36_cu101_pyt181
          python_version: '3.6'
          pytorch_version: 1.8.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py36_cu102_pyt181
          python_version: '3.6'
          pytorch_version: 1.8.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu111
          name: linux_conda_py36_cu111_pyt181
          python_version: '3.6'
          pytorch_version: 1.8.1
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
368
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
369
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
370
371
          cu_version: cu92
          name: linux_conda_py37_cu92_pyt14
facebook-github-bot's avatar
facebook-github-bot committed
372
373
374
          python_version: '3.7'
          pytorch_version: '1.4'
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
375
          context: DOCKERHUB_TOKEN
facebook-github-bot's avatar
facebook-github-bot committed
376
          cu_version: cu101
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
377
          name: linux_conda_py37_cu101_pyt14
facebook-github-bot's avatar
facebook-github-bot committed
378
379
380
          python_version: '3.7'
          pytorch_version: '1.4'
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
381
          context: DOCKERHUB_TOKEN
facebook-github-bot's avatar
facebook-github-bot committed
382
          cu_version: cu92
383
          name: linux_conda_py37_cu92_pyt150
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
384
          python_version: '3.7'
385
          pytorch_version: 1.5.0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
386
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
387
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
388
          cu_version: cu101
389
          name: linux_conda_py37_cu101_pyt150
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
390
          python_version: '3.7'
391
          pytorch_version: 1.5.0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
392
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
393
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
394
          cu_version: cu102
395
          name: linux_conda_py37_cu102_pyt150
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
396
          python_version: '3.7'
397
398
          pytorch_version: 1.5.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
399
          context: DOCKERHUB_TOKEN
400
401
402
403
404
          cu_version: cu92
          name: linux_conda_py37_cu92_pyt151
          python_version: '3.7'
          pytorch_version: 1.5.1
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
405
          context: DOCKERHUB_TOKEN
406
407
408
409
410
          cu_version: cu101
          name: linux_conda_py37_cu101_pyt151
          python_version: '3.7'
          pytorch_version: 1.5.1
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
411
          context: DOCKERHUB_TOKEN
412
413
414
415
          cu_version: cu102
          name: linux_conda_py37_cu102_pyt151
          python_version: '3.7'
          pytorch_version: 1.5.1
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
416
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
417
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
418
419
420
421
422
          cu_version: cu92
          name: linux_conda_py37_cu92_pyt160
          python_version: '3.7'
          pytorch_version: 1.6.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
423
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
424
425
426
427
428
          cu_version: cu101
          name: linux_conda_py37_cu101_pyt160
          python_version: '3.7'
          pytorch_version: 1.6.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
429
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
430
431
432
433
          cu_version: cu102
          name: linux_conda_py37_cu102_pyt160
          python_version: '3.7'
          pytorch_version: 1.6.0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
434
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
435
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
436
437
438
439
440
          cu_version: cu101
          name: linux_conda_py37_cu101_pyt170
          python_version: '3.7'
          pytorch_version: 1.7.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
441
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
442
443
444
445
446
          cu_version: cu102
          name: linux_conda_py37_cu102_pyt170
          python_version: '3.7'
          pytorch_version: 1.7.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
447
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
448
449
450
451
          cu_version: cu110
          name: linux_conda_py37_cu110_pyt170
          python_version: '3.7'
          pytorch_version: 1.7.0
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py37_cu101_pyt171
          python_version: '3.7'
          pytorch_version: 1.7.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py37_cu102_pyt171
          python_version: '3.7'
          pytorch_version: 1.7.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu110
          name: linux_conda_py37_cu110_pyt171
          python_version: '3.7'
          pytorch_version: 1.7.1
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py37_cu101_pyt180
          python_version: '3.7'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py37_cu102_pyt180
          python_version: '3.7'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu111
          name: linux_conda_py37_cu111_pyt180
          python_version: '3.7'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py37_cu101_pyt181
          python_version: '3.7'
          pytorch_version: 1.8.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py37_cu102_pyt181
          python_version: '3.7'
          pytorch_version: 1.8.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu111
          name: linux_conda_py37_cu111_pyt181
          python_version: '3.7'
          pytorch_version: 1.8.1
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
506
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
507
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
508
509
          cu_version: cu92
          name: linux_conda_py38_cu92_pyt14
facebook-github-bot's avatar
facebook-github-bot committed
510
511
512
          python_version: '3.8'
          pytorch_version: '1.4'
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
513
          context: DOCKERHUB_TOKEN
facebook-github-bot's avatar
facebook-github-bot committed
514
          cu_version: cu101
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
515
          name: linux_conda_py38_cu101_pyt14
facebook-github-bot's avatar
facebook-github-bot committed
516
517
          python_version: '3.8'
          pytorch_version: '1.4'
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
518
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
519
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
520
          cu_version: cu92
521
          name: linux_conda_py38_cu92_pyt150
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
522
          python_version: '3.8'
523
          pytorch_version: 1.5.0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
524
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
525
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
526
          cu_version: cu101
527
          name: linux_conda_py38_cu101_pyt150
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
528
          python_version: '3.8'
529
          pytorch_version: 1.5.0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
530
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
531
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
532
          cu_version: cu102
533
          name: linux_conda_py38_cu102_pyt150
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
534
          python_version: '3.8'
535
536
          pytorch_version: 1.5.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
537
          context: DOCKERHUB_TOKEN
538
539
540
541
542
          cu_version: cu92
          name: linux_conda_py38_cu92_pyt151
          python_version: '3.8'
          pytorch_version: 1.5.1
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
543
          context: DOCKERHUB_TOKEN
544
545
546
547
548
          cu_version: cu101
          name: linux_conda_py38_cu101_pyt151
          python_version: '3.8'
          pytorch_version: 1.5.1
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
549
          context: DOCKERHUB_TOKEN
550
551
552
553
          cu_version: cu102
          name: linux_conda_py38_cu102_pyt151
          python_version: '3.8'
          pytorch_version: 1.5.1
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
554
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
555
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
556
557
558
559
560
          cu_version: cu92
          name: linux_conda_py38_cu92_pyt160
          python_version: '3.8'
          pytorch_version: 1.6.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
561
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
562
563
564
565
566
          cu_version: cu101
          name: linux_conda_py38_cu101_pyt160
          python_version: '3.8'
          pytorch_version: 1.6.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
567
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
568
569
570
571
          cu_version: cu102
          name: linux_conda_py38_cu102_pyt160
          python_version: '3.8'
          pytorch_version: 1.6.0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
572
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
573
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
574
575
576
577
578
          cu_version: cu101
          name: linux_conda_py38_cu101_pyt170
          python_version: '3.8'
          pytorch_version: 1.7.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
579
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
580
581
582
583
584
          cu_version: cu102
          name: linux_conda_py38_cu102_pyt170
          python_version: '3.8'
          pytorch_version: 1.7.0
      - binary_linux_conda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
585
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
586
587
588
589
          cu_version: cu110
          name: linux_conda_py38_cu110_pyt170
          python_version: '3.8'
          pytorch_version: 1.7.0
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py38_cu101_pyt171
          python_version: '3.8'
          pytorch_version: 1.7.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py38_cu102_pyt171
          python_version: '3.8'
          pytorch_version: 1.7.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu110
          name: linux_conda_py38_cu110_pyt171
          python_version: '3.8'
          pytorch_version: 1.7.1
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py38_cu101_pyt180
          python_version: '3.8'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py38_cu102_pyt180
          python_version: '3.8'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu111
          name: linux_conda_py38_cu111_pyt180
          python_version: '3.8'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py38_cu101_pyt181
          python_version: '3.8'
          pytorch_version: 1.8.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py38_cu102_pyt181
          python_version: '3.8'
          pytorch_version: 1.8.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu111
          name: linux_conda_py38_cu111_pyt181
          python_version: '3.8'
          pytorch_version: 1.8.1
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py39_cu101_pyt171
          python_version: '3.9'
          pytorch_version: 1.7.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py39_cu102_pyt171
          python_version: '3.9'
          pytorch_version: 1.7.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu110
          name: linux_conda_py39_cu110_pyt171
          python_version: '3.9'
          pytorch_version: 1.7.1
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py39_cu101_pyt180
          python_version: '3.9'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py39_cu102_pyt180
          python_version: '3.9'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu111
          name: linux_conda_py39_cu111_pyt180
          python_version: '3.9'
          pytorch_version: 1.8.0
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu101
          name: linux_conda_py39_cu101_pyt181
          python_version: '3.9'
          pytorch_version: 1.8.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu102
          name: linux_conda_py39_cu102_pyt181
          python_version: '3.9'
          pytorch_version: 1.8.1
      - binary_linux_conda:
          context: DOCKERHUB_TOKEN
          cu_version: cu111
          name: linux_conda_py39_cu111_pyt181
          python_version: '3.9'
          pytorch_version: 1.8.1
facebook-github-bot's avatar
facebook-github-bot committed
698
      - binary_linux_conda_cuda:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
699
700
701
          name: testrun_conda_cuda_py36_cu101_pyt14
          context: DOCKERHUB_TOKEN
          python_version: "3.6"
facebook-github-bot's avatar
facebook-github-bot committed
702
          pytorch_version: "1.4"
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
703
          cu_version: "cu101"
704
705
706
707
708
709
      - binary_linux_conda_cuda:
          name: testrun_conda_cuda_py37_cu102_pyt160
          context: DOCKERHUB_TOKEN
          python_version: "3.7"
          pytorch_version: '1.6.0'
          cu_version: "cu102"
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
710
711
      - binary_linux_conda_cuda:
          name: testrun_conda_cuda_py37_cu110_pyt170
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
712
          context: DOCKERHUB_TOKEN
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
713
714
715
          python_version: "3.7"
          pytorch_version: '1.7.0'
          cu_version: "cu110"
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
716
717
718
719
720
721
      - binary_linux_conda_cuda:
          name: testrun_conda_cuda_py39_cu111_pyt181
          context: DOCKERHUB_TOKEN
          python_version: "3.9"
          pytorch_version: '1.8.1'
          cu_version: "cu111"
722
723
      - binary_macos_wheel:
          cu_version: cpu
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
724
          name: macos_wheel_py36_cpu
725
          python_version: '3.6'
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
726
          pytorch_version: '1.7.1'
727
728
      - binary_macos_wheel:
          cu_version: cpu
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
729
          name: macos_wheel_py37_cpu
730
          python_version: '3.7'
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
731
          pytorch_version: '1.7.1'
732
733
      - binary_macos_wheel:
          cu_version: cpu
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
734
          name: macos_wheel_py38_cpu
735
          python_version: '3.8'
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
736
737
738
739
740
741
          pytorch_version: '1.7.1'
      - binary_macos_wheel:
          cu_version: cpu
          name: macos_wheel_py39_cpu
          python_version: '3.9'
          pytorch_version: '1.7.1'