From 3f428d9981099ea0560dbed4893bff4785718d08 Mon Sep 17 00:00:00 2001 From: Nicholas Dahm Date: Wed, 11 Sep 2024 15:09:43 -0700 Subject: [PATCH] pytorch 2.4.0 + 2.4.1 Summary: Apparently pytorch 2.4 is now supported as per [this closed issue](https://github.com/facebookresearch/pytorch3d/issues/1863). Added the `2.4.0` & `2.4.1` versions to `regenerate.py` then ran that as per the `README_fb.md` which generated `config.yml` changes. Reviewed By: bottler Differential Revision: D62517831 fbshipit-source-id: 002e276dfe2fa078136ff2f6c747d937abbadd1a --- .circleci/config.yml | 140 ++++++++++++++++++++++++++++++++++++++++ .circleci/regenerate.py | 2 + 2 files changed, 142 insertions(+) diff --git a/.circleci/config.yml b/.circleci/config.yml index 2519f39f..58be87c9 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -274,6 +274,34 @@ workflows: name: linux_conda_py38_cu121_pyt231 python_version: '3.8' pytorch_version: 2.3.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py38_cu118_pyt240 + python_version: '3.8' + pytorch_version: 2.4.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py38_cu121_pyt240 + python_version: '3.8' + pytorch_version: 2.4.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py38_cu118_pyt241 + python_version: '3.8' + pytorch_version: 2.4.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py38_cu121_pyt241 + python_version: '3.8' + pytorch_version: 2.4.1 - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cuda117 context: DOCKERHUB_TOKEN @@ -386,6 +414,34 @@ workflows: name: linux_conda_py39_cu121_pyt231 python_version: '3.9' pytorch_version: 2.3.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py39_cu118_pyt240 + python_version: '3.9' + pytorch_version: 2.4.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py39_cu121_pyt240 + python_version: '3.9' + pytorch_version: 2.4.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py39_cu118_pyt241 + python_version: '3.9' + pytorch_version: 2.4.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py39_cu121_pyt241 + python_version: '3.9' + pytorch_version: 2.4.1 - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cuda117 context: DOCKERHUB_TOKEN @@ -498,6 +554,34 @@ workflows: name: linux_conda_py310_cu121_pyt231 python_version: '3.10' pytorch_version: 2.3.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py310_cu118_pyt240 + python_version: '3.10' + pytorch_version: 2.4.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py310_cu121_pyt240 + python_version: '3.10' + pytorch_version: 2.4.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py310_cu118_pyt241 + python_version: '3.10' + pytorch_version: 2.4.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py310_cu121_pyt241 + python_version: '3.10' + pytorch_version: 2.4.1 - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cuda118 context: DOCKERHUB_TOKEN @@ -582,6 +666,34 @@ workflows: name: linux_conda_py311_cu121_pyt231 python_version: '3.11' pytorch_version: 2.3.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py311_cu118_pyt240 + python_version: '3.11' + pytorch_version: 2.4.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py311_cu121_pyt240 + python_version: '3.11' + pytorch_version: 2.4.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py311_cu118_pyt241 + python_version: '3.11' + pytorch_version: 2.4.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py311_cu121_pyt241 + python_version: '3.11' + pytorch_version: 2.4.1 - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cuda118 context: DOCKERHUB_TOKEN @@ -624,6 +736,34 @@ workflows: name: linux_conda_py312_cu121_pyt231 python_version: '3.12' pytorch_version: 2.3.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py312_cu118_pyt240 + python_version: '3.12' + pytorch_version: 2.4.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py312_cu121_pyt240 + python_version: '3.12' + pytorch_version: 2.4.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py312_cu118_pyt241 + python_version: '3.12' + pytorch_version: 2.4.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py312_cu121_pyt241 + python_version: '3.12' + pytorch_version: 2.4.1 - binary_linux_conda_cuda: name: testrun_conda_cuda_py310_cu117_pyt201 context: DOCKERHUB_TOKEN diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index 5e7d6257..0208de22 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -27,6 +27,8 @@ CONDA_CUDA_VERSIONS = { "2.2.0": ["cu118", "cu121"], "2.2.2": ["cu118", "cu121"], "2.3.1": ["cu118", "cu121"], + "2.4.0": ["cu118", "cu121"], + "2.4.1": ["cu118", "cu121"], }