From 302da69461b372a3f35c11214c9c826ebd2b336e Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Mon, 17 Jun 2024 05:57:59 -0700 Subject: [PATCH] builds for PyTorch 2.2.1 2.2.2 2.3.0 2.3.1 Summary: Build for new pytorch versions Reviewed By: MichaelRamamonjisoa Differential Revision: D58668956 fbshipit-source-id: 7fdfb377b370448d6147daded6a21b8db87586fb --- .circleci/config.yml | 140 ++++++++++++++++++++++++++++++++++++++++ .circleci/regenerate.py | 2 + INSTALL.md | 2 +- 3 files changed, 143 insertions(+), 1 deletion(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 17e36959..8cc5bb52 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -302,6 +302,34 @@ workflows: name: linux_conda_py38_cu121_pyt220 python_version: '3.8' pytorch_version: 2.2.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py38_cu118_pyt222 + python_version: '3.8' + pytorch_version: 2.2.2 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py38_cu121_pyt222 + python_version: '3.8' + pytorch_version: 2.2.2 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py38_cu118_pyt231 + python_version: '3.8' + pytorch_version: 2.3.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py38_cu121_pyt231 + python_version: '3.8' + pytorch_version: 2.3.1 - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cuda113 context: DOCKERHUB_TOKEN @@ -442,6 +470,34 @@ workflows: name: linux_conda_py39_cu121_pyt220 python_version: '3.9' pytorch_version: 2.2.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py39_cu118_pyt222 + python_version: '3.9' + pytorch_version: 2.2.2 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py39_cu121_pyt222 + python_version: '3.9' + pytorch_version: 2.2.2 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py39_cu118_pyt231 + python_version: '3.9' + pytorch_version: 2.3.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py39_cu121_pyt231 + python_version: '3.9' + pytorch_version: 2.3.1 - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cuda113 context: DOCKERHUB_TOKEN @@ -582,6 +638,34 @@ workflows: name: linux_conda_py310_cu121_pyt220 python_version: '3.10' pytorch_version: 2.2.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py310_cu118_pyt222 + python_version: '3.10' + pytorch_version: 2.2.2 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py310_cu121_pyt222 + python_version: '3.10' + pytorch_version: 2.2.2 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py310_cu118_pyt231 + python_version: '3.10' + pytorch_version: 2.3.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + 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 @@ -638,6 +722,34 @@ workflows: name: linux_conda_py311_cu121_pyt220 python_version: '3.11' pytorch_version: 2.2.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py311_cu118_pyt222 + python_version: '3.11' + pytorch_version: 2.2.2 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py311_cu121_pyt222 + python_version: '3.11' + pytorch_version: 2.2.2 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py311_cu118_pyt231 + python_version: '3.11' + pytorch_version: 2.3.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + 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 @@ -652,6 +764,34 @@ workflows: name: linux_conda_py312_cu121_pyt220 python_version: '3.12' pytorch_version: 2.2.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py312_cu118_pyt222 + python_version: '3.12' + pytorch_version: 2.2.2 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py312_cu121_pyt222 + python_version: '3.12' + pytorch_version: 2.2.2 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py312_cu118_pyt231 + python_version: '3.12' + pytorch_version: 2.3.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda121 + context: DOCKERHUB_TOKEN + cu_version: cu121 + name: linux_conda_py312_cu121_pyt231 + python_version: '3.12' + pytorch_version: 2.3.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 6fe9e5bf..4a90e92c 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -29,6 +29,8 @@ "2.1.1": ["cu118", "cu121"], "2.1.2": ["cu118", "cu121"], "2.2.0": ["cu118", "cu121"], + "2.2.2": ["cu118", "cu121"], + "2.3.1": ["cu118", "cu121"], } diff --git a/INSTALL.md b/INSTALL.md index 5439a4ed..5e491e51 100644 --- a/INSTALL.md +++ b/INSTALL.md @@ -9,7 +9,7 @@ The core library is written in PyTorch. Several components have underlying imple - Linux or macOS or Windows - Python 3.8, 3.9 or 3.10 -- PyTorch 1.12.0, 1.12.1, 1.13.0, 2.0.0, 2.0.1, 2.1.0, 2.1.1, 2.1.2 or 2.2.0. +- PyTorch 1.12.0, 1.12.1, 1.13.0, 2.0.0, 2.0.1, 2.1.0, 2.1.1, 2.1.2, 2.2.0, 2.2.1, 2.2.2, 2.3.0 or 2.3.1. - torchvision that matches the PyTorch installation. You can install them together as explained at pytorch.org to make sure of this. - gcc & g++ ≥ 4.9 - [fvcore](https://github.com/facebookresearch/fvcore)