1# @generated DO NOT EDIT MANUALLY 2 3# Template is at: .github/templates/linux_binary_build_workflow.yml.j2 4# Generation script: .github/scripts/generate_ci_workflows.py 5name: linux-binary-conda 6 7 8on: 9 push: 10 # NOTE: Meta Employees can trigger new nightlies using: https://fburl.com/trigger_pytorch_nightly_build 11 branches: 12 - nightly 13 tags: 14 # NOTE: Binary build pipelines should only get triggered on release candidate builds 15 # Release candidate tags look like: v1.11.0-rc1 16 - v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+ 17 - 'ciflow/binaries/*' 18 - 'ciflow/binaries_conda/*' 19 workflow_dispatch: 20 21env: 22 # Needed for conda builds 23 ALPINE_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/alpine" 24 ANACONDA_USER: pytorch 25 AWS_DEFAULT_REGION: us-east-1 26 BINARY_ENV_FILE: /tmp/env 27 BUILD_ENVIRONMENT: linux-binary-conda 28 BUILDER_ROOT: /builder 29 GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} 30 PR_NUMBER: ${{ github.event.pull_request.number }} 31 PYTORCH_FINAL_PACKAGE_DIR: /artifacts 32 PYTORCH_ROOT: /pytorch 33 SHA1: ${{ github.event.pull_request.head.sha || github.sha }} 34 SKIP_ALL_TESTS: 0 35concurrency: 36 group: linux-binary-conda-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }} 37 cancel-in-progress: true 38 39jobs: 40 conda-py3_8-cpu-build: 41 if: ${{ github.repository_owner == 'pytorch' }} 42 uses: ./.github/workflows/_binary-build-linux.yml 43 with: 44 PYTORCH_ROOT: /pytorch 45 BUILDER_ROOT: /builder 46 PACKAGE_TYPE: conda 47 # TODO: This is a legacy variable that we eventually want to get rid of in 48 # favor of GPU_ARCH_VERSION 49 DESIRED_CUDA: cpu 50 GPU_ARCH_TYPE: cpu 51 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 52 DESIRED_PYTHON: "3.8" 53 build_name: conda-py3_8-cpu 54 build_environment: linux-binary-conda 55 secrets: 56 github-token: ${{ secrets.GITHUB_TOKEN }} 57 conda-py3_8-cpu-test: # Testing 58 if: ${{ github.repository_owner == 'pytorch' }} 59 needs: conda-py3_8-cpu-build 60 uses: ./.github/workflows/_binary-test-linux.yml 61 with: 62 PYTORCH_ROOT: /pytorch 63 BUILDER_ROOT: /builder 64 PACKAGE_TYPE: conda 65 # TODO: This is a legacy variable that we eventually want to get rid of in 66 # favor of GPU_ARCH_VERSION 67 DESIRED_CUDA: cpu 68 GPU_ARCH_TYPE: cpu 69 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 70 DESIRED_PYTHON: "3.8" 71 build_name: conda-py3_8-cpu 72 build_environment: linux-binary-conda 73 runs_on: linux.4xlarge 74 secrets: 75 github-token: ${{ secrets.GITHUB_TOKEN }} 76 conda-py3_8-cpu-upload: # Uploading 77 if: ${{ github.repository_owner == 'pytorch' }} 78 permissions: 79 id-token: write 80 contents: read 81 needs: conda-py3_8-cpu-test 82 with: 83 PYTORCH_ROOT: /pytorch 84 BUILDER_ROOT: /builder 85 PACKAGE_TYPE: conda 86 # TODO: This is a legacy variable that we eventually want to get rid of in 87 # favor of GPU_ARCH_VERSION 88 DESIRED_CUDA: cpu 89 GPU_ARCH_TYPE: cpu 90 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 91 DESIRED_PYTHON: "3.8" 92 build_name: conda-py3_8-cpu 93 secrets: 94 github-token: ${{ secrets.GITHUB_TOKEN }} 95 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 96 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 97 uses: ./.github/workflows/_binary-upload.yml 98 99 conda-py3_8-cuda11_8-build: 100 if: ${{ github.repository_owner == 'pytorch' }} 101 uses: ./.github/workflows/_binary-build-linux.yml 102 with: 103 PYTORCH_ROOT: /pytorch 104 BUILDER_ROOT: /builder 105 PACKAGE_TYPE: conda 106 # TODO: This is a legacy variable that we eventually want to get rid of in 107 # favor of GPU_ARCH_VERSION 108 DESIRED_CUDA: cu118 109 GPU_ARCH_VERSION: 11.8 110 GPU_ARCH_TYPE: cuda 111 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 112 DESIRED_PYTHON: "3.8" 113 runs_on: linux.24xlarge.ephemeral 114 build_name: conda-py3_8-cuda11_8 115 build_environment: linux-binary-conda 116 secrets: 117 github-token: ${{ secrets.GITHUB_TOKEN }} 118 conda-py3_8-cuda11_8-test: # Testing 119 if: ${{ github.repository_owner == 'pytorch' }} 120 needs: conda-py3_8-cuda11_8-build 121 uses: ./.github/workflows/_binary-test-linux.yml 122 with: 123 PYTORCH_ROOT: /pytorch 124 BUILDER_ROOT: /builder 125 PACKAGE_TYPE: conda 126 # TODO: This is a legacy variable that we eventually want to get rid of in 127 # favor of GPU_ARCH_VERSION 128 DESIRED_CUDA: cu118 129 GPU_ARCH_VERSION: 11.8 130 GPU_ARCH_TYPE: cuda 131 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 132 DESIRED_PYTHON: "3.8" 133 build_name: conda-py3_8-cuda11_8 134 build_environment: linux-binary-conda 135 runs_on: linux.4xlarge.nvidia.gpu 136 secrets: 137 github-token: ${{ secrets.GITHUB_TOKEN }} 138 conda-py3_8-cuda11_8-upload: # Uploading 139 if: ${{ github.repository_owner == 'pytorch' }} 140 permissions: 141 id-token: write 142 contents: read 143 needs: conda-py3_8-cuda11_8-test 144 with: 145 PYTORCH_ROOT: /pytorch 146 BUILDER_ROOT: /builder 147 PACKAGE_TYPE: conda 148 # TODO: This is a legacy variable that we eventually want to get rid of in 149 # favor of GPU_ARCH_VERSION 150 DESIRED_CUDA: cu118 151 GPU_ARCH_VERSION: 11.8 152 GPU_ARCH_TYPE: cuda 153 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 154 DESIRED_PYTHON: "3.8" 155 build_name: conda-py3_8-cuda11_8 156 secrets: 157 github-token: ${{ secrets.GITHUB_TOKEN }} 158 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 159 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 160 uses: ./.github/workflows/_binary-upload.yml 161 162 conda-py3_8-cuda12_1-build: 163 if: ${{ github.repository_owner == 'pytorch' }} 164 uses: ./.github/workflows/_binary-build-linux.yml 165 with: 166 PYTORCH_ROOT: /pytorch 167 BUILDER_ROOT: /builder 168 PACKAGE_TYPE: conda 169 # TODO: This is a legacy variable that we eventually want to get rid of in 170 # favor of GPU_ARCH_VERSION 171 DESIRED_CUDA: cu121 172 GPU_ARCH_VERSION: 12.1 173 GPU_ARCH_TYPE: cuda 174 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 175 DESIRED_PYTHON: "3.8" 176 runs_on: linux.24xlarge.ephemeral 177 build_name: conda-py3_8-cuda12_1 178 build_environment: linux-binary-conda 179 secrets: 180 github-token: ${{ secrets.GITHUB_TOKEN }} 181 conda-py3_8-cuda12_1-test: # Testing 182 if: ${{ github.repository_owner == 'pytorch' }} 183 needs: conda-py3_8-cuda12_1-build 184 uses: ./.github/workflows/_binary-test-linux.yml 185 with: 186 PYTORCH_ROOT: /pytorch 187 BUILDER_ROOT: /builder 188 PACKAGE_TYPE: conda 189 # TODO: This is a legacy variable that we eventually want to get rid of in 190 # favor of GPU_ARCH_VERSION 191 DESIRED_CUDA: cu121 192 GPU_ARCH_VERSION: 12.1 193 GPU_ARCH_TYPE: cuda 194 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 195 DESIRED_PYTHON: "3.8" 196 build_name: conda-py3_8-cuda12_1 197 build_environment: linux-binary-conda 198 runs_on: linux.4xlarge.nvidia.gpu 199 secrets: 200 github-token: ${{ secrets.GITHUB_TOKEN }} 201 conda-py3_8-cuda12_1-upload: # Uploading 202 if: ${{ github.repository_owner == 'pytorch' }} 203 permissions: 204 id-token: write 205 contents: read 206 needs: conda-py3_8-cuda12_1-test 207 with: 208 PYTORCH_ROOT: /pytorch 209 BUILDER_ROOT: /builder 210 PACKAGE_TYPE: conda 211 # TODO: This is a legacy variable that we eventually want to get rid of in 212 # favor of GPU_ARCH_VERSION 213 DESIRED_CUDA: cu121 214 GPU_ARCH_VERSION: 12.1 215 GPU_ARCH_TYPE: cuda 216 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 217 DESIRED_PYTHON: "3.8" 218 build_name: conda-py3_8-cuda12_1 219 secrets: 220 github-token: ${{ secrets.GITHUB_TOKEN }} 221 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 222 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 223 uses: ./.github/workflows/_binary-upload.yml 224 225 conda-py3_8-cuda12_4-build: 226 if: ${{ github.repository_owner == 'pytorch' }} 227 uses: ./.github/workflows/_binary-build-linux.yml 228 with: 229 PYTORCH_ROOT: /pytorch 230 BUILDER_ROOT: /builder 231 PACKAGE_TYPE: conda 232 # TODO: This is a legacy variable that we eventually want to get rid of in 233 # favor of GPU_ARCH_VERSION 234 DESIRED_CUDA: cu124 235 GPU_ARCH_VERSION: 12.4 236 GPU_ARCH_TYPE: cuda 237 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 238 DESIRED_PYTHON: "3.8" 239 runs_on: linux.24xlarge.ephemeral 240 build_name: conda-py3_8-cuda12_4 241 build_environment: linux-binary-conda 242 secrets: 243 github-token: ${{ secrets.GITHUB_TOKEN }} 244 conda-py3_8-cuda12_4-test: # Testing 245 if: ${{ github.repository_owner == 'pytorch' }} 246 needs: conda-py3_8-cuda12_4-build 247 uses: ./.github/workflows/_binary-test-linux.yml 248 with: 249 PYTORCH_ROOT: /pytorch 250 BUILDER_ROOT: /builder 251 PACKAGE_TYPE: conda 252 # TODO: This is a legacy variable that we eventually want to get rid of in 253 # favor of GPU_ARCH_VERSION 254 DESIRED_CUDA: cu124 255 GPU_ARCH_VERSION: 12.4 256 GPU_ARCH_TYPE: cuda 257 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 258 DESIRED_PYTHON: "3.8" 259 build_name: conda-py3_8-cuda12_4 260 build_environment: linux-binary-conda 261 runs_on: linux.4xlarge.nvidia.gpu 262 secrets: 263 github-token: ${{ secrets.GITHUB_TOKEN }} 264 conda-py3_8-cuda12_4-upload: # Uploading 265 if: ${{ github.repository_owner == 'pytorch' }} 266 permissions: 267 id-token: write 268 contents: read 269 needs: conda-py3_8-cuda12_4-test 270 with: 271 PYTORCH_ROOT: /pytorch 272 BUILDER_ROOT: /builder 273 PACKAGE_TYPE: conda 274 # TODO: This is a legacy variable that we eventually want to get rid of in 275 # favor of GPU_ARCH_VERSION 276 DESIRED_CUDA: cu124 277 GPU_ARCH_VERSION: 12.4 278 GPU_ARCH_TYPE: cuda 279 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 280 DESIRED_PYTHON: "3.8" 281 build_name: conda-py3_8-cuda12_4 282 secrets: 283 github-token: ${{ secrets.GITHUB_TOKEN }} 284 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 285 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 286 uses: ./.github/workflows/_binary-upload.yml 287 288 conda-py3_9-cpu-build: 289 if: ${{ github.repository_owner == 'pytorch' }} 290 uses: ./.github/workflows/_binary-build-linux.yml 291 with: 292 PYTORCH_ROOT: /pytorch 293 BUILDER_ROOT: /builder 294 PACKAGE_TYPE: conda 295 # TODO: This is a legacy variable that we eventually want to get rid of in 296 # favor of GPU_ARCH_VERSION 297 DESIRED_CUDA: cpu 298 GPU_ARCH_TYPE: cpu 299 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 300 DESIRED_PYTHON: "3.9" 301 build_name: conda-py3_9-cpu 302 build_environment: linux-binary-conda 303 secrets: 304 github-token: ${{ secrets.GITHUB_TOKEN }} 305 conda-py3_9-cpu-test: # Testing 306 if: ${{ github.repository_owner == 'pytorch' }} 307 needs: conda-py3_9-cpu-build 308 uses: ./.github/workflows/_binary-test-linux.yml 309 with: 310 PYTORCH_ROOT: /pytorch 311 BUILDER_ROOT: /builder 312 PACKAGE_TYPE: conda 313 # TODO: This is a legacy variable that we eventually want to get rid of in 314 # favor of GPU_ARCH_VERSION 315 DESIRED_CUDA: cpu 316 GPU_ARCH_TYPE: cpu 317 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 318 DESIRED_PYTHON: "3.9" 319 build_name: conda-py3_9-cpu 320 build_environment: linux-binary-conda 321 runs_on: linux.4xlarge 322 secrets: 323 github-token: ${{ secrets.GITHUB_TOKEN }} 324 conda-py3_9-cpu-upload: # Uploading 325 if: ${{ github.repository_owner == 'pytorch' }} 326 permissions: 327 id-token: write 328 contents: read 329 needs: conda-py3_9-cpu-test 330 with: 331 PYTORCH_ROOT: /pytorch 332 BUILDER_ROOT: /builder 333 PACKAGE_TYPE: conda 334 # TODO: This is a legacy variable that we eventually want to get rid of in 335 # favor of GPU_ARCH_VERSION 336 DESIRED_CUDA: cpu 337 GPU_ARCH_TYPE: cpu 338 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 339 DESIRED_PYTHON: "3.9" 340 build_name: conda-py3_9-cpu 341 secrets: 342 github-token: ${{ secrets.GITHUB_TOKEN }} 343 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 344 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 345 uses: ./.github/workflows/_binary-upload.yml 346 347 conda-py3_9-cuda11_8-build: 348 if: ${{ github.repository_owner == 'pytorch' }} 349 uses: ./.github/workflows/_binary-build-linux.yml 350 with: 351 PYTORCH_ROOT: /pytorch 352 BUILDER_ROOT: /builder 353 PACKAGE_TYPE: conda 354 # TODO: This is a legacy variable that we eventually want to get rid of in 355 # favor of GPU_ARCH_VERSION 356 DESIRED_CUDA: cu118 357 GPU_ARCH_VERSION: 11.8 358 GPU_ARCH_TYPE: cuda 359 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 360 DESIRED_PYTHON: "3.9" 361 runs_on: linux.24xlarge.ephemeral 362 build_name: conda-py3_9-cuda11_8 363 build_environment: linux-binary-conda 364 secrets: 365 github-token: ${{ secrets.GITHUB_TOKEN }} 366 conda-py3_9-cuda11_8-test: # Testing 367 if: ${{ github.repository_owner == 'pytorch' }} 368 needs: conda-py3_9-cuda11_8-build 369 uses: ./.github/workflows/_binary-test-linux.yml 370 with: 371 PYTORCH_ROOT: /pytorch 372 BUILDER_ROOT: /builder 373 PACKAGE_TYPE: conda 374 # TODO: This is a legacy variable that we eventually want to get rid of in 375 # favor of GPU_ARCH_VERSION 376 DESIRED_CUDA: cu118 377 GPU_ARCH_VERSION: 11.8 378 GPU_ARCH_TYPE: cuda 379 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 380 DESIRED_PYTHON: "3.9" 381 build_name: conda-py3_9-cuda11_8 382 build_environment: linux-binary-conda 383 runs_on: linux.4xlarge.nvidia.gpu 384 secrets: 385 github-token: ${{ secrets.GITHUB_TOKEN }} 386 conda-py3_9-cuda11_8-upload: # Uploading 387 if: ${{ github.repository_owner == 'pytorch' }} 388 permissions: 389 id-token: write 390 contents: read 391 needs: conda-py3_9-cuda11_8-test 392 with: 393 PYTORCH_ROOT: /pytorch 394 BUILDER_ROOT: /builder 395 PACKAGE_TYPE: conda 396 # TODO: This is a legacy variable that we eventually want to get rid of in 397 # favor of GPU_ARCH_VERSION 398 DESIRED_CUDA: cu118 399 GPU_ARCH_VERSION: 11.8 400 GPU_ARCH_TYPE: cuda 401 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 402 DESIRED_PYTHON: "3.9" 403 build_name: conda-py3_9-cuda11_8 404 secrets: 405 github-token: ${{ secrets.GITHUB_TOKEN }} 406 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 407 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 408 uses: ./.github/workflows/_binary-upload.yml 409 410 conda-py3_9-cuda12_1-build: 411 if: ${{ github.repository_owner == 'pytorch' }} 412 uses: ./.github/workflows/_binary-build-linux.yml 413 with: 414 PYTORCH_ROOT: /pytorch 415 BUILDER_ROOT: /builder 416 PACKAGE_TYPE: conda 417 # TODO: This is a legacy variable that we eventually want to get rid of in 418 # favor of GPU_ARCH_VERSION 419 DESIRED_CUDA: cu121 420 GPU_ARCH_VERSION: 12.1 421 GPU_ARCH_TYPE: cuda 422 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 423 DESIRED_PYTHON: "3.9" 424 runs_on: linux.24xlarge.ephemeral 425 build_name: conda-py3_9-cuda12_1 426 build_environment: linux-binary-conda 427 secrets: 428 github-token: ${{ secrets.GITHUB_TOKEN }} 429 conda-py3_9-cuda12_1-test: # Testing 430 if: ${{ github.repository_owner == 'pytorch' }} 431 needs: conda-py3_9-cuda12_1-build 432 uses: ./.github/workflows/_binary-test-linux.yml 433 with: 434 PYTORCH_ROOT: /pytorch 435 BUILDER_ROOT: /builder 436 PACKAGE_TYPE: conda 437 # TODO: This is a legacy variable that we eventually want to get rid of in 438 # favor of GPU_ARCH_VERSION 439 DESIRED_CUDA: cu121 440 GPU_ARCH_VERSION: 12.1 441 GPU_ARCH_TYPE: cuda 442 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 443 DESIRED_PYTHON: "3.9" 444 build_name: conda-py3_9-cuda12_1 445 build_environment: linux-binary-conda 446 runs_on: linux.4xlarge.nvidia.gpu 447 secrets: 448 github-token: ${{ secrets.GITHUB_TOKEN }} 449 conda-py3_9-cuda12_1-upload: # Uploading 450 if: ${{ github.repository_owner == 'pytorch' }} 451 permissions: 452 id-token: write 453 contents: read 454 needs: conda-py3_9-cuda12_1-test 455 with: 456 PYTORCH_ROOT: /pytorch 457 BUILDER_ROOT: /builder 458 PACKAGE_TYPE: conda 459 # TODO: This is a legacy variable that we eventually want to get rid of in 460 # favor of GPU_ARCH_VERSION 461 DESIRED_CUDA: cu121 462 GPU_ARCH_VERSION: 12.1 463 GPU_ARCH_TYPE: cuda 464 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 465 DESIRED_PYTHON: "3.9" 466 build_name: conda-py3_9-cuda12_1 467 secrets: 468 github-token: ${{ secrets.GITHUB_TOKEN }} 469 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 470 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 471 uses: ./.github/workflows/_binary-upload.yml 472 473 conda-py3_9-cuda12_4-build: 474 if: ${{ github.repository_owner == 'pytorch' }} 475 uses: ./.github/workflows/_binary-build-linux.yml 476 with: 477 PYTORCH_ROOT: /pytorch 478 BUILDER_ROOT: /builder 479 PACKAGE_TYPE: conda 480 # TODO: This is a legacy variable that we eventually want to get rid of in 481 # favor of GPU_ARCH_VERSION 482 DESIRED_CUDA: cu124 483 GPU_ARCH_VERSION: 12.4 484 GPU_ARCH_TYPE: cuda 485 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 486 DESIRED_PYTHON: "3.9" 487 runs_on: linux.24xlarge.ephemeral 488 build_name: conda-py3_9-cuda12_4 489 build_environment: linux-binary-conda 490 secrets: 491 github-token: ${{ secrets.GITHUB_TOKEN }} 492 conda-py3_9-cuda12_4-test: # Testing 493 if: ${{ github.repository_owner == 'pytorch' }} 494 needs: conda-py3_9-cuda12_4-build 495 uses: ./.github/workflows/_binary-test-linux.yml 496 with: 497 PYTORCH_ROOT: /pytorch 498 BUILDER_ROOT: /builder 499 PACKAGE_TYPE: conda 500 # TODO: This is a legacy variable that we eventually want to get rid of in 501 # favor of GPU_ARCH_VERSION 502 DESIRED_CUDA: cu124 503 GPU_ARCH_VERSION: 12.4 504 GPU_ARCH_TYPE: cuda 505 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 506 DESIRED_PYTHON: "3.9" 507 build_name: conda-py3_9-cuda12_4 508 build_environment: linux-binary-conda 509 runs_on: linux.4xlarge.nvidia.gpu 510 secrets: 511 github-token: ${{ secrets.GITHUB_TOKEN }} 512 conda-py3_9-cuda12_4-upload: # Uploading 513 if: ${{ github.repository_owner == 'pytorch' }} 514 permissions: 515 id-token: write 516 contents: read 517 needs: conda-py3_9-cuda12_4-test 518 with: 519 PYTORCH_ROOT: /pytorch 520 BUILDER_ROOT: /builder 521 PACKAGE_TYPE: conda 522 # TODO: This is a legacy variable that we eventually want to get rid of in 523 # favor of GPU_ARCH_VERSION 524 DESIRED_CUDA: cu124 525 GPU_ARCH_VERSION: 12.4 526 GPU_ARCH_TYPE: cuda 527 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 528 DESIRED_PYTHON: "3.9" 529 build_name: conda-py3_9-cuda12_4 530 secrets: 531 github-token: ${{ secrets.GITHUB_TOKEN }} 532 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 533 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 534 uses: ./.github/workflows/_binary-upload.yml 535 536 conda-py3_10-cpu-build: 537 if: ${{ github.repository_owner == 'pytorch' }} 538 uses: ./.github/workflows/_binary-build-linux.yml 539 with: 540 PYTORCH_ROOT: /pytorch 541 BUILDER_ROOT: /builder 542 PACKAGE_TYPE: conda 543 # TODO: This is a legacy variable that we eventually want to get rid of in 544 # favor of GPU_ARCH_VERSION 545 DESIRED_CUDA: cpu 546 GPU_ARCH_TYPE: cpu 547 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 548 DESIRED_PYTHON: "3.10" 549 build_name: conda-py3_10-cpu 550 build_environment: linux-binary-conda 551 secrets: 552 github-token: ${{ secrets.GITHUB_TOKEN }} 553 conda-py3_10-cpu-test: # Testing 554 if: ${{ github.repository_owner == 'pytorch' }} 555 needs: conda-py3_10-cpu-build 556 uses: ./.github/workflows/_binary-test-linux.yml 557 with: 558 PYTORCH_ROOT: /pytorch 559 BUILDER_ROOT: /builder 560 PACKAGE_TYPE: conda 561 # TODO: This is a legacy variable that we eventually want to get rid of in 562 # favor of GPU_ARCH_VERSION 563 DESIRED_CUDA: cpu 564 GPU_ARCH_TYPE: cpu 565 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 566 DESIRED_PYTHON: "3.10" 567 build_name: conda-py3_10-cpu 568 build_environment: linux-binary-conda 569 runs_on: linux.4xlarge 570 secrets: 571 github-token: ${{ secrets.GITHUB_TOKEN }} 572 conda-py3_10-cpu-upload: # Uploading 573 if: ${{ github.repository_owner == 'pytorch' }} 574 permissions: 575 id-token: write 576 contents: read 577 needs: conda-py3_10-cpu-test 578 with: 579 PYTORCH_ROOT: /pytorch 580 BUILDER_ROOT: /builder 581 PACKAGE_TYPE: conda 582 # TODO: This is a legacy variable that we eventually want to get rid of in 583 # favor of GPU_ARCH_VERSION 584 DESIRED_CUDA: cpu 585 GPU_ARCH_TYPE: cpu 586 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 587 DESIRED_PYTHON: "3.10" 588 build_name: conda-py3_10-cpu 589 secrets: 590 github-token: ${{ secrets.GITHUB_TOKEN }} 591 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 592 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 593 uses: ./.github/workflows/_binary-upload.yml 594 595 conda-py3_10-cuda11_8-build: 596 if: ${{ github.repository_owner == 'pytorch' }} 597 uses: ./.github/workflows/_binary-build-linux.yml 598 with: 599 PYTORCH_ROOT: /pytorch 600 BUILDER_ROOT: /builder 601 PACKAGE_TYPE: conda 602 # TODO: This is a legacy variable that we eventually want to get rid of in 603 # favor of GPU_ARCH_VERSION 604 DESIRED_CUDA: cu118 605 GPU_ARCH_VERSION: 11.8 606 GPU_ARCH_TYPE: cuda 607 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 608 DESIRED_PYTHON: "3.10" 609 runs_on: linux.24xlarge.ephemeral 610 build_name: conda-py3_10-cuda11_8 611 build_environment: linux-binary-conda 612 secrets: 613 github-token: ${{ secrets.GITHUB_TOKEN }} 614 conda-py3_10-cuda11_8-test: # Testing 615 if: ${{ github.repository_owner == 'pytorch' }} 616 needs: conda-py3_10-cuda11_8-build 617 uses: ./.github/workflows/_binary-test-linux.yml 618 with: 619 PYTORCH_ROOT: /pytorch 620 BUILDER_ROOT: /builder 621 PACKAGE_TYPE: conda 622 # TODO: This is a legacy variable that we eventually want to get rid of in 623 # favor of GPU_ARCH_VERSION 624 DESIRED_CUDA: cu118 625 GPU_ARCH_VERSION: 11.8 626 GPU_ARCH_TYPE: cuda 627 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 628 DESIRED_PYTHON: "3.10" 629 build_name: conda-py3_10-cuda11_8 630 build_environment: linux-binary-conda 631 runs_on: linux.4xlarge.nvidia.gpu 632 secrets: 633 github-token: ${{ secrets.GITHUB_TOKEN }} 634 conda-py3_10-cuda11_8-upload: # Uploading 635 if: ${{ github.repository_owner == 'pytorch' }} 636 permissions: 637 id-token: write 638 contents: read 639 needs: conda-py3_10-cuda11_8-test 640 with: 641 PYTORCH_ROOT: /pytorch 642 BUILDER_ROOT: /builder 643 PACKAGE_TYPE: conda 644 # TODO: This is a legacy variable that we eventually want to get rid of in 645 # favor of GPU_ARCH_VERSION 646 DESIRED_CUDA: cu118 647 GPU_ARCH_VERSION: 11.8 648 GPU_ARCH_TYPE: cuda 649 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 650 DESIRED_PYTHON: "3.10" 651 build_name: conda-py3_10-cuda11_8 652 secrets: 653 github-token: ${{ secrets.GITHUB_TOKEN }} 654 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 655 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 656 uses: ./.github/workflows/_binary-upload.yml 657 658 conda-py3_10-cuda12_1-build: 659 if: ${{ github.repository_owner == 'pytorch' }} 660 uses: ./.github/workflows/_binary-build-linux.yml 661 with: 662 PYTORCH_ROOT: /pytorch 663 BUILDER_ROOT: /builder 664 PACKAGE_TYPE: conda 665 # TODO: This is a legacy variable that we eventually want to get rid of in 666 # favor of GPU_ARCH_VERSION 667 DESIRED_CUDA: cu121 668 GPU_ARCH_VERSION: 12.1 669 GPU_ARCH_TYPE: cuda 670 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 671 DESIRED_PYTHON: "3.10" 672 runs_on: linux.24xlarge.ephemeral 673 build_name: conda-py3_10-cuda12_1 674 build_environment: linux-binary-conda 675 secrets: 676 github-token: ${{ secrets.GITHUB_TOKEN }} 677 conda-py3_10-cuda12_1-test: # Testing 678 if: ${{ github.repository_owner == 'pytorch' }} 679 needs: conda-py3_10-cuda12_1-build 680 uses: ./.github/workflows/_binary-test-linux.yml 681 with: 682 PYTORCH_ROOT: /pytorch 683 BUILDER_ROOT: /builder 684 PACKAGE_TYPE: conda 685 # TODO: This is a legacy variable that we eventually want to get rid of in 686 # favor of GPU_ARCH_VERSION 687 DESIRED_CUDA: cu121 688 GPU_ARCH_VERSION: 12.1 689 GPU_ARCH_TYPE: cuda 690 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 691 DESIRED_PYTHON: "3.10" 692 build_name: conda-py3_10-cuda12_1 693 build_environment: linux-binary-conda 694 runs_on: linux.4xlarge.nvidia.gpu 695 secrets: 696 github-token: ${{ secrets.GITHUB_TOKEN }} 697 conda-py3_10-cuda12_1-upload: # Uploading 698 if: ${{ github.repository_owner == 'pytorch' }} 699 permissions: 700 id-token: write 701 contents: read 702 needs: conda-py3_10-cuda12_1-test 703 with: 704 PYTORCH_ROOT: /pytorch 705 BUILDER_ROOT: /builder 706 PACKAGE_TYPE: conda 707 # TODO: This is a legacy variable that we eventually want to get rid of in 708 # favor of GPU_ARCH_VERSION 709 DESIRED_CUDA: cu121 710 GPU_ARCH_VERSION: 12.1 711 GPU_ARCH_TYPE: cuda 712 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 713 DESIRED_PYTHON: "3.10" 714 build_name: conda-py3_10-cuda12_1 715 secrets: 716 github-token: ${{ secrets.GITHUB_TOKEN }} 717 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 718 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 719 uses: ./.github/workflows/_binary-upload.yml 720 721 conda-py3_10-cuda12_4-build: 722 if: ${{ github.repository_owner == 'pytorch' }} 723 uses: ./.github/workflows/_binary-build-linux.yml 724 with: 725 PYTORCH_ROOT: /pytorch 726 BUILDER_ROOT: /builder 727 PACKAGE_TYPE: conda 728 # TODO: This is a legacy variable that we eventually want to get rid of in 729 # favor of GPU_ARCH_VERSION 730 DESIRED_CUDA: cu124 731 GPU_ARCH_VERSION: 12.4 732 GPU_ARCH_TYPE: cuda 733 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 734 DESIRED_PYTHON: "3.10" 735 runs_on: linux.24xlarge.ephemeral 736 build_name: conda-py3_10-cuda12_4 737 build_environment: linux-binary-conda 738 secrets: 739 github-token: ${{ secrets.GITHUB_TOKEN }} 740 conda-py3_10-cuda12_4-test: # Testing 741 if: ${{ github.repository_owner == 'pytorch' }} 742 needs: conda-py3_10-cuda12_4-build 743 uses: ./.github/workflows/_binary-test-linux.yml 744 with: 745 PYTORCH_ROOT: /pytorch 746 BUILDER_ROOT: /builder 747 PACKAGE_TYPE: conda 748 # TODO: This is a legacy variable that we eventually want to get rid of in 749 # favor of GPU_ARCH_VERSION 750 DESIRED_CUDA: cu124 751 GPU_ARCH_VERSION: 12.4 752 GPU_ARCH_TYPE: cuda 753 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 754 DESIRED_PYTHON: "3.10" 755 build_name: conda-py3_10-cuda12_4 756 build_environment: linux-binary-conda 757 runs_on: linux.4xlarge.nvidia.gpu 758 secrets: 759 github-token: ${{ secrets.GITHUB_TOKEN }} 760 conda-py3_10-cuda12_4-upload: # Uploading 761 if: ${{ github.repository_owner == 'pytorch' }} 762 permissions: 763 id-token: write 764 contents: read 765 needs: conda-py3_10-cuda12_4-test 766 with: 767 PYTORCH_ROOT: /pytorch 768 BUILDER_ROOT: /builder 769 PACKAGE_TYPE: conda 770 # TODO: This is a legacy variable that we eventually want to get rid of in 771 # favor of GPU_ARCH_VERSION 772 DESIRED_CUDA: cu124 773 GPU_ARCH_VERSION: 12.4 774 GPU_ARCH_TYPE: cuda 775 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 776 DESIRED_PYTHON: "3.10" 777 build_name: conda-py3_10-cuda12_4 778 secrets: 779 github-token: ${{ secrets.GITHUB_TOKEN }} 780 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 781 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 782 uses: ./.github/workflows/_binary-upload.yml 783 784 conda-py3_11-cpu-build: 785 if: ${{ github.repository_owner == 'pytorch' }} 786 uses: ./.github/workflows/_binary-build-linux.yml 787 with: 788 PYTORCH_ROOT: /pytorch 789 BUILDER_ROOT: /builder 790 PACKAGE_TYPE: conda 791 # TODO: This is a legacy variable that we eventually want to get rid of in 792 # favor of GPU_ARCH_VERSION 793 DESIRED_CUDA: cpu 794 GPU_ARCH_TYPE: cpu 795 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 796 DESIRED_PYTHON: "3.11" 797 build_name: conda-py3_11-cpu 798 build_environment: linux-binary-conda 799 secrets: 800 github-token: ${{ secrets.GITHUB_TOKEN }} 801 conda-py3_11-cpu-test: # Testing 802 if: ${{ github.repository_owner == 'pytorch' }} 803 needs: conda-py3_11-cpu-build 804 uses: ./.github/workflows/_binary-test-linux.yml 805 with: 806 PYTORCH_ROOT: /pytorch 807 BUILDER_ROOT: /builder 808 PACKAGE_TYPE: conda 809 # TODO: This is a legacy variable that we eventually want to get rid of in 810 # favor of GPU_ARCH_VERSION 811 DESIRED_CUDA: cpu 812 GPU_ARCH_TYPE: cpu 813 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 814 DESIRED_PYTHON: "3.11" 815 build_name: conda-py3_11-cpu 816 build_environment: linux-binary-conda 817 runs_on: linux.4xlarge 818 secrets: 819 github-token: ${{ secrets.GITHUB_TOKEN }} 820 conda-py3_11-cpu-upload: # Uploading 821 if: ${{ github.repository_owner == 'pytorch' }} 822 permissions: 823 id-token: write 824 contents: read 825 needs: conda-py3_11-cpu-test 826 with: 827 PYTORCH_ROOT: /pytorch 828 BUILDER_ROOT: /builder 829 PACKAGE_TYPE: conda 830 # TODO: This is a legacy variable that we eventually want to get rid of in 831 # favor of GPU_ARCH_VERSION 832 DESIRED_CUDA: cpu 833 GPU_ARCH_TYPE: cpu 834 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 835 DESIRED_PYTHON: "3.11" 836 build_name: conda-py3_11-cpu 837 secrets: 838 github-token: ${{ secrets.GITHUB_TOKEN }} 839 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 840 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 841 uses: ./.github/workflows/_binary-upload.yml 842 843 conda-py3_11-cuda11_8-build: 844 if: ${{ github.repository_owner == 'pytorch' }} 845 uses: ./.github/workflows/_binary-build-linux.yml 846 with: 847 PYTORCH_ROOT: /pytorch 848 BUILDER_ROOT: /builder 849 PACKAGE_TYPE: conda 850 # TODO: This is a legacy variable that we eventually want to get rid of in 851 # favor of GPU_ARCH_VERSION 852 DESIRED_CUDA: cu118 853 GPU_ARCH_VERSION: 11.8 854 GPU_ARCH_TYPE: cuda 855 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 856 DESIRED_PYTHON: "3.11" 857 runs_on: linux.24xlarge.ephemeral 858 build_name: conda-py3_11-cuda11_8 859 build_environment: linux-binary-conda 860 secrets: 861 github-token: ${{ secrets.GITHUB_TOKEN }} 862 conda-py3_11-cuda11_8-test: # Testing 863 if: ${{ github.repository_owner == 'pytorch' }} 864 needs: conda-py3_11-cuda11_8-build 865 uses: ./.github/workflows/_binary-test-linux.yml 866 with: 867 PYTORCH_ROOT: /pytorch 868 BUILDER_ROOT: /builder 869 PACKAGE_TYPE: conda 870 # TODO: This is a legacy variable that we eventually want to get rid of in 871 # favor of GPU_ARCH_VERSION 872 DESIRED_CUDA: cu118 873 GPU_ARCH_VERSION: 11.8 874 GPU_ARCH_TYPE: cuda 875 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 876 DESIRED_PYTHON: "3.11" 877 build_name: conda-py3_11-cuda11_8 878 build_environment: linux-binary-conda 879 runs_on: linux.4xlarge.nvidia.gpu 880 secrets: 881 github-token: ${{ secrets.GITHUB_TOKEN }} 882 conda-py3_11-cuda11_8-upload: # Uploading 883 if: ${{ github.repository_owner == 'pytorch' }} 884 permissions: 885 id-token: write 886 contents: read 887 needs: conda-py3_11-cuda11_8-test 888 with: 889 PYTORCH_ROOT: /pytorch 890 BUILDER_ROOT: /builder 891 PACKAGE_TYPE: conda 892 # TODO: This is a legacy variable that we eventually want to get rid of in 893 # favor of GPU_ARCH_VERSION 894 DESIRED_CUDA: cu118 895 GPU_ARCH_VERSION: 11.8 896 GPU_ARCH_TYPE: cuda 897 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 898 DESIRED_PYTHON: "3.11" 899 build_name: conda-py3_11-cuda11_8 900 secrets: 901 github-token: ${{ secrets.GITHUB_TOKEN }} 902 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 903 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 904 uses: ./.github/workflows/_binary-upload.yml 905 906 conda-py3_11-cuda12_1-build: 907 if: ${{ github.repository_owner == 'pytorch' }} 908 uses: ./.github/workflows/_binary-build-linux.yml 909 with: 910 PYTORCH_ROOT: /pytorch 911 BUILDER_ROOT: /builder 912 PACKAGE_TYPE: conda 913 # TODO: This is a legacy variable that we eventually want to get rid of in 914 # favor of GPU_ARCH_VERSION 915 DESIRED_CUDA: cu121 916 GPU_ARCH_VERSION: 12.1 917 GPU_ARCH_TYPE: cuda 918 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 919 DESIRED_PYTHON: "3.11" 920 runs_on: linux.24xlarge.ephemeral 921 build_name: conda-py3_11-cuda12_1 922 build_environment: linux-binary-conda 923 secrets: 924 github-token: ${{ secrets.GITHUB_TOKEN }} 925 conda-py3_11-cuda12_1-test: # Testing 926 if: ${{ github.repository_owner == 'pytorch' }} 927 needs: conda-py3_11-cuda12_1-build 928 uses: ./.github/workflows/_binary-test-linux.yml 929 with: 930 PYTORCH_ROOT: /pytorch 931 BUILDER_ROOT: /builder 932 PACKAGE_TYPE: conda 933 # TODO: This is a legacy variable that we eventually want to get rid of in 934 # favor of GPU_ARCH_VERSION 935 DESIRED_CUDA: cu121 936 GPU_ARCH_VERSION: 12.1 937 GPU_ARCH_TYPE: cuda 938 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 939 DESIRED_PYTHON: "3.11" 940 build_name: conda-py3_11-cuda12_1 941 build_environment: linux-binary-conda 942 runs_on: linux.4xlarge.nvidia.gpu 943 secrets: 944 github-token: ${{ secrets.GITHUB_TOKEN }} 945 conda-py3_11-cuda12_1-upload: # Uploading 946 if: ${{ github.repository_owner == 'pytorch' }} 947 permissions: 948 id-token: write 949 contents: read 950 needs: conda-py3_11-cuda12_1-test 951 with: 952 PYTORCH_ROOT: /pytorch 953 BUILDER_ROOT: /builder 954 PACKAGE_TYPE: conda 955 # TODO: This is a legacy variable that we eventually want to get rid of in 956 # favor of GPU_ARCH_VERSION 957 DESIRED_CUDA: cu121 958 GPU_ARCH_VERSION: 12.1 959 GPU_ARCH_TYPE: cuda 960 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 961 DESIRED_PYTHON: "3.11" 962 build_name: conda-py3_11-cuda12_1 963 secrets: 964 github-token: ${{ secrets.GITHUB_TOKEN }} 965 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 966 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 967 uses: ./.github/workflows/_binary-upload.yml 968 969 conda-py3_11-cuda12_4-build: 970 if: ${{ github.repository_owner == 'pytorch' }} 971 uses: ./.github/workflows/_binary-build-linux.yml 972 with: 973 PYTORCH_ROOT: /pytorch 974 BUILDER_ROOT: /builder 975 PACKAGE_TYPE: conda 976 # TODO: This is a legacy variable that we eventually want to get rid of in 977 # favor of GPU_ARCH_VERSION 978 DESIRED_CUDA: cu124 979 GPU_ARCH_VERSION: 12.4 980 GPU_ARCH_TYPE: cuda 981 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 982 DESIRED_PYTHON: "3.11" 983 runs_on: linux.24xlarge.ephemeral 984 build_name: conda-py3_11-cuda12_4 985 build_environment: linux-binary-conda 986 secrets: 987 github-token: ${{ secrets.GITHUB_TOKEN }} 988 conda-py3_11-cuda12_4-test: # Testing 989 if: ${{ github.repository_owner == 'pytorch' }} 990 needs: conda-py3_11-cuda12_4-build 991 uses: ./.github/workflows/_binary-test-linux.yml 992 with: 993 PYTORCH_ROOT: /pytorch 994 BUILDER_ROOT: /builder 995 PACKAGE_TYPE: conda 996 # TODO: This is a legacy variable that we eventually want to get rid of in 997 # favor of GPU_ARCH_VERSION 998 DESIRED_CUDA: cu124 999 GPU_ARCH_VERSION: 12.4 1000 GPU_ARCH_TYPE: cuda 1001 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 1002 DESIRED_PYTHON: "3.11" 1003 build_name: conda-py3_11-cuda12_4 1004 build_environment: linux-binary-conda 1005 runs_on: linux.4xlarge.nvidia.gpu 1006 secrets: 1007 github-token: ${{ secrets.GITHUB_TOKEN }} 1008 conda-py3_11-cuda12_4-upload: # Uploading 1009 if: ${{ github.repository_owner == 'pytorch' }} 1010 permissions: 1011 id-token: write 1012 contents: read 1013 needs: conda-py3_11-cuda12_4-test 1014 with: 1015 PYTORCH_ROOT: /pytorch 1016 BUILDER_ROOT: /builder 1017 PACKAGE_TYPE: conda 1018 # TODO: This is a legacy variable that we eventually want to get rid of in 1019 # favor of GPU_ARCH_VERSION 1020 DESIRED_CUDA: cu124 1021 GPU_ARCH_VERSION: 12.4 1022 GPU_ARCH_TYPE: cuda 1023 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 1024 DESIRED_PYTHON: "3.11" 1025 build_name: conda-py3_11-cuda12_4 1026 secrets: 1027 github-token: ${{ secrets.GITHUB_TOKEN }} 1028 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 1029 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 1030 uses: ./.github/workflows/_binary-upload.yml 1031 1032 conda-py3_12-cpu-build: 1033 if: ${{ github.repository_owner == 'pytorch' }} 1034 uses: ./.github/workflows/_binary-build-linux.yml 1035 with: 1036 PYTORCH_ROOT: /pytorch 1037 BUILDER_ROOT: /builder 1038 PACKAGE_TYPE: conda 1039 # TODO: This is a legacy variable that we eventually want to get rid of in 1040 # favor of GPU_ARCH_VERSION 1041 DESIRED_CUDA: cpu 1042 GPU_ARCH_TYPE: cpu 1043 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 1044 DESIRED_PYTHON: "3.12" 1045 build_name: conda-py3_12-cpu 1046 build_environment: linux-binary-conda 1047 secrets: 1048 github-token: ${{ secrets.GITHUB_TOKEN }} 1049 conda-py3_12-cpu-test: # Testing 1050 if: ${{ github.repository_owner == 'pytorch' }} 1051 needs: conda-py3_12-cpu-build 1052 uses: ./.github/workflows/_binary-test-linux.yml 1053 with: 1054 PYTORCH_ROOT: /pytorch 1055 BUILDER_ROOT: /builder 1056 PACKAGE_TYPE: conda 1057 # TODO: This is a legacy variable that we eventually want to get rid of in 1058 # favor of GPU_ARCH_VERSION 1059 DESIRED_CUDA: cpu 1060 GPU_ARCH_TYPE: cpu 1061 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 1062 DESIRED_PYTHON: "3.12" 1063 build_name: conda-py3_12-cpu 1064 build_environment: linux-binary-conda 1065 runs_on: linux.4xlarge 1066 secrets: 1067 github-token: ${{ secrets.GITHUB_TOKEN }} 1068 conda-py3_12-cpu-upload: # Uploading 1069 if: ${{ github.repository_owner == 'pytorch' }} 1070 permissions: 1071 id-token: write 1072 contents: read 1073 needs: conda-py3_12-cpu-test 1074 with: 1075 PYTORCH_ROOT: /pytorch 1076 BUILDER_ROOT: /builder 1077 PACKAGE_TYPE: conda 1078 # TODO: This is a legacy variable that we eventually want to get rid of in 1079 # favor of GPU_ARCH_VERSION 1080 DESIRED_CUDA: cpu 1081 GPU_ARCH_TYPE: cpu 1082 DOCKER_IMAGE: pytorch/conda-builder:cpu-2.4 1083 DESIRED_PYTHON: "3.12" 1084 build_name: conda-py3_12-cpu 1085 secrets: 1086 github-token: ${{ secrets.GITHUB_TOKEN }} 1087 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 1088 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 1089 uses: ./.github/workflows/_binary-upload.yml 1090 1091 conda-py3_12-cuda11_8-build: 1092 if: ${{ github.repository_owner == 'pytorch' }} 1093 uses: ./.github/workflows/_binary-build-linux.yml 1094 with: 1095 PYTORCH_ROOT: /pytorch 1096 BUILDER_ROOT: /builder 1097 PACKAGE_TYPE: conda 1098 # TODO: This is a legacy variable that we eventually want to get rid of in 1099 # favor of GPU_ARCH_VERSION 1100 DESIRED_CUDA: cu118 1101 GPU_ARCH_VERSION: 11.8 1102 GPU_ARCH_TYPE: cuda 1103 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 1104 DESIRED_PYTHON: "3.12" 1105 runs_on: linux.24xlarge.ephemeral 1106 build_name: conda-py3_12-cuda11_8 1107 build_environment: linux-binary-conda 1108 secrets: 1109 github-token: ${{ secrets.GITHUB_TOKEN }} 1110 conda-py3_12-cuda11_8-test: # Testing 1111 if: ${{ github.repository_owner == 'pytorch' }} 1112 needs: conda-py3_12-cuda11_8-build 1113 uses: ./.github/workflows/_binary-test-linux.yml 1114 with: 1115 PYTORCH_ROOT: /pytorch 1116 BUILDER_ROOT: /builder 1117 PACKAGE_TYPE: conda 1118 # TODO: This is a legacy variable that we eventually want to get rid of in 1119 # favor of GPU_ARCH_VERSION 1120 DESIRED_CUDA: cu118 1121 GPU_ARCH_VERSION: 11.8 1122 GPU_ARCH_TYPE: cuda 1123 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 1124 DESIRED_PYTHON: "3.12" 1125 build_name: conda-py3_12-cuda11_8 1126 build_environment: linux-binary-conda 1127 runs_on: linux.4xlarge.nvidia.gpu 1128 secrets: 1129 github-token: ${{ secrets.GITHUB_TOKEN }} 1130 conda-py3_12-cuda11_8-upload: # Uploading 1131 if: ${{ github.repository_owner == 'pytorch' }} 1132 permissions: 1133 id-token: write 1134 contents: read 1135 needs: conda-py3_12-cuda11_8-test 1136 with: 1137 PYTORCH_ROOT: /pytorch 1138 BUILDER_ROOT: /builder 1139 PACKAGE_TYPE: conda 1140 # TODO: This is a legacy variable that we eventually want to get rid of in 1141 # favor of GPU_ARCH_VERSION 1142 DESIRED_CUDA: cu118 1143 GPU_ARCH_VERSION: 11.8 1144 GPU_ARCH_TYPE: cuda 1145 DOCKER_IMAGE: pytorch/conda-builder:cuda11.8-2.4 1146 DESIRED_PYTHON: "3.12" 1147 build_name: conda-py3_12-cuda11_8 1148 secrets: 1149 github-token: ${{ secrets.GITHUB_TOKEN }} 1150 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 1151 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 1152 uses: ./.github/workflows/_binary-upload.yml 1153 1154 conda-py3_12-cuda12_1-build: 1155 if: ${{ github.repository_owner == 'pytorch' }} 1156 uses: ./.github/workflows/_binary-build-linux.yml 1157 with: 1158 PYTORCH_ROOT: /pytorch 1159 BUILDER_ROOT: /builder 1160 PACKAGE_TYPE: conda 1161 # TODO: This is a legacy variable that we eventually want to get rid of in 1162 # favor of GPU_ARCH_VERSION 1163 DESIRED_CUDA: cu121 1164 GPU_ARCH_VERSION: 12.1 1165 GPU_ARCH_TYPE: cuda 1166 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 1167 DESIRED_PYTHON: "3.12" 1168 runs_on: linux.24xlarge.ephemeral 1169 build_name: conda-py3_12-cuda12_1 1170 build_environment: linux-binary-conda 1171 secrets: 1172 github-token: ${{ secrets.GITHUB_TOKEN }} 1173 conda-py3_12-cuda12_1-test: # Testing 1174 if: ${{ github.repository_owner == 'pytorch' }} 1175 needs: conda-py3_12-cuda12_1-build 1176 uses: ./.github/workflows/_binary-test-linux.yml 1177 with: 1178 PYTORCH_ROOT: /pytorch 1179 BUILDER_ROOT: /builder 1180 PACKAGE_TYPE: conda 1181 # TODO: This is a legacy variable that we eventually want to get rid of in 1182 # favor of GPU_ARCH_VERSION 1183 DESIRED_CUDA: cu121 1184 GPU_ARCH_VERSION: 12.1 1185 GPU_ARCH_TYPE: cuda 1186 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 1187 DESIRED_PYTHON: "3.12" 1188 build_name: conda-py3_12-cuda12_1 1189 build_environment: linux-binary-conda 1190 runs_on: linux.4xlarge.nvidia.gpu 1191 secrets: 1192 github-token: ${{ secrets.GITHUB_TOKEN }} 1193 conda-py3_12-cuda12_1-upload: # Uploading 1194 if: ${{ github.repository_owner == 'pytorch' }} 1195 permissions: 1196 id-token: write 1197 contents: read 1198 needs: conda-py3_12-cuda12_1-test 1199 with: 1200 PYTORCH_ROOT: /pytorch 1201 BUILDER_ROOT: /builder 1202 PACKAGE_TYPE: conda 1203 # TODO: This is a legacy variable that we eventually want to get rid of in 1204 # favor of GPU_ARCH_VERSION 1205 DESIRED_CUDA: cu121 1206 GPU_ARCH_VERSION: 12.1 1207 GPU_ARCH_TYPE: cuda 1208 DOCKER_IMAGE: pytorch/conda-builder:cuda12.1-2.4 1209 DESIRED_PYTHON: "3.12" 1210 build_name: conda-py3_12-cuda12_1 1211 secrets: 1212 github-token: ${{ secrets.GITHUB_TOKEN }} 1213 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 1214 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 1215 uses: ./.github/workflows/_binary-upload.yml 1216 1217 conda-py3_12-cuda12_4-build: 1218 if: ${{ github.repository_owner == 'pytorch' }} 1219 uses: ./.github/workflows/_binary-build-linux.yml 1220 with: 1221 PYTORCH_ROOT: /pytorch 1222 BUILDER_ROOT: /builder 1223 PACKAGE_TYPE: conda 1224 # TODO: This is a legacy variable that we eventually want to get rid of in 1225 # favor of GPU_ARCH_VERSION 1226 DESIRED_CUDA: cu124 1227 GPU_ARCH_VERSION: 12.4 1228 GPU_ARCH_TYPE: cuda 1229 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 1230 DESIRED_PYTHON: "3.12" 1231 runs_on: linux.24xlarge.ephemeral 1232 build_name: conda-py3_12-cuda12_4 1233 build_environment: linux-binary-conda 1234 secrets: 1235 github-token: ${{ secrets.GITHUB_TOKEN }} 1236 conda-py3_12-cuda12_4-test: # Testing 1237 if: ${{ github.repository_owner == 'pytorch' }} 1238 needs: conda-py3_12-cuda12_4-build 1239 uses: ./.github/workflows/_binary-test-linux.yml 1240 with: 1241 PYTORCH_ROOT: /pytorch 1242 BUILDER_ROOT: /builder 1243 PACKAGE_TYPE: conda 1244 # TODO: This is a legacy variable that we eventually want to get rid of in 1245 # favor of GPU_ARCH_VERSION 1246 DESIRED_CUDA: cu124 1247 GPU_ARCH_VERSION: 12.4 1248 GPU_ARCH_TYPE: cuda 1249 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 1250 DESIRED_PYTHON: "3.12" 1251 build_name: conda-py3_12-cuda12_4 1252 build_environment: linux-binary-conda 1253 runs_on: linux.4xlarge.nvidia.gpu 1254 secrets: 1255 github-token: ${{ secrets.GITHUB_TOKEN }} 1256 conda-py3_12-cuda12_4-upload: # Uploading 1257 if: ${{ github.repository_owner == 'pytorch' }} 1258 permissions: 1259 id-token: write 1260 contents: read 1261 needs: conda-py3_12-cuda12_4-test 1262 with: 1263 PYTORCH_ROOT: /pytorch 1264 BUILDER_ROOT: /builder 1265 PACKAGE_TYPE: conda 1266 # TODO: This is a legacy variable that we eventually want to get rid of in 1267 # favor of GPU_ARCH_VERSION 1268 DESIRED_CUDA: cu124 1269 GPU_ARCH_VERSION: 12.4 1270 GPU_ARCH_TYPE: cuda 1271 DOCKER_IMAGE: pytorch/conda-builder:cuda12.4-2.4 1272 DESIRED_PYTHON: "3.12" 1273 build_name: conda-py3_12-cuda12_4 1274 secrets: 1275 github-token: ${{ secrets.GITHUB_TOKEN }} 1276 conda-pytorchbot-token: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} 1277 conda-pytorchbot-token-test: ${{ secrets.CONDA_PYTORCHBOT_TOKEN_TEST }} 1278 uses: ./.github/workflows/_binary-upload.yml 1279