From 062e6c54aea9b8d9882b90fbfdfb62bcd4e946d1 Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Wed, 10 May 2023 02:38:47 -0700 Subject: [PATCH] builds for PyTorch 2.0.1; drop 1.9 Summary: Drop support for PyTorch 1.9.0 and 1.9.1. Reviewed By: shapovalov Differential Revision: D45704329 fbshipit-source-id: c0fe3ecf6a1eb9bcd4163785c0cb4bf4f5060f50 --- .circleci/config.yml | 82 +++++++++++++++++++---------------------- .circleci/regenerate.py | 3 +- INSTALL.md | 2 +- 3 files changed, 40 insertions(+), 47 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index d016eb88..0fe3eb29 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -180,30 +180,6 @@ workflows: jobs: # - main: # context: DOCKERHUB_TOKEN - - binary_linux_conda: - context: DOCKERHUB_TOKEN - cu_version: cu102 - name: linux_conda_py38_cu102_pyt190 - python_version: '3.8' - pytorch_version: 1.9.0 - - binary_linux_conda: - context: DOCKERHUB_TOKEN - cu_version: cu111 - name: linux_conda_py38_cu111_pyt190 - python_version: '3.8' - pytorch_version: 1.9.0 - - binary_linux_conda: - context: DOCKERHUB_TOKEN - cu_version: cu102 - name: linux_conda_py38_cu102_pyt191 - python_version: '3.8' - pytorch_version: 1.9.1 - - binary_linux_conda: - context: DOCKERHUB_TOKEN - cu_version: cu111 - name: linux_conda_py38_cu111_pyt191 - python_version: '3.8' - pytorch_version: 1.9.1 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu102 @@ -370,29 +346,19 @@ workflows: python_version: '3.8' pytorch_version: 2.0.0 - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda117 context: DOCKERHUB_TOKEN - cu_version: cu102 - name: linux_conda_py39_cu102_pyt190 - python_version: '3.9' - pytorch_version: 1.9.0 + cu_version: cu117 + name: linux_conda_py38_cu117_pyt201 + python_version: '3.8' + pytorch_version: 2.0.1 - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 context: DOCKERHUB_TOKEN - cu_version: cu111 - name: linux_conda_py39_cu111_pyt190 - python_version: '3.9' - pytorch_version: 1.9.0 - - binary_linux_conda: - context: DOCKERHUB_TOKEN - cu_version: cu102 - name: linux_conda_py39_cu102_pyt191 - python_version: '3.9' - pytorch_version: 1.9.1 - - binary_linux_conda: - context: DOCKERHUB_TOKEN - cu_version: cu111 - name: linux_conda_py39_cu111_pyt191 - python_version: '3.9' - pytorch_version: 1.9.1 + cu_version: cu118 + name: linux_conda_py38_cu118_pyt201 + python_version: '3.8' + pytorch_version: 2.0.1 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu102 @@ -558,6 +524,20 @@ workflows: name: linux_conda_py39_cu118_pyt200 python_version: '3.9' pytorch_version: 2.0.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda117 + context: DOCKERHUB_TOKEN + cu_version: cu117 + name: linux_conda_py39_cu117_pyt201 + python_version: '3.9' + pytorch_version: 2.0.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py39_cu118_pyt201 + python_version: '3.9' + pytorch_version: 2.0.1 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu102 @@ -666,6 +646,20 @@ workflows: name: linux_conda_py310_cu118_pyt200 python_version: '3.10' pytorch_version: 2.0.0 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda117 + context: DOCKERHUB_TOKEN + cu_version: cu117 + name: linux_conda_py310_cu117_pyt201 + python_version: '3.10' + pytorch_version: 2.0.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda118 + context: DOCKERHUB_TOKEN + cu_version: cu118 + name: linux_conda_py310_cu118_pyt201 + python_version: '3.10' + pytorch_version: 2.0.1 - binary_linux_conda_cuda: name: testrun_conda_cuda_py38_cu102_pyt190 context: DOCKERHUB_TOKEN diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index bf449bd1..c434df35 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -20,8 +20,6 @@ from packaging import version # version of pytorch. # Pytorch 1.4 also supports cuda 10.0 but we no longer build for cuda 10.0 at all. CONDA_CUDA_VERSIONS = { - "1.9.0": ["cu102", "cu111"], - "1.9.1": ["cu102", "cu111"], "1.10.0": ["cu102", "cu111", "cu113"], "1.10.1": ["cu102", "cu111", "cu113"], "1.10.2": ["cu102", "cu111", "cu113"], @@ -31,6 +29,7 @@ CONDA_CUDA_VERSIONS = { "1.13.0": ["cu116", "cu117"], "1.13.1": ["cu116", "cu117"], "2.0.0": ["cu117", "cu118"], + "2.0.1": ["cu117", "cu118"], } diff --git a/INSTALL.md b/INSTALL.md index bbc11afa..f7e34a73 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.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0, 1.12.0, 1.12.1, 1.13.0 or 2.0.0. +- PyTorch 1.10.0, 1.10.1, 1.10.2, 1.11.0, 1.12.0, 1.12.1, 1.13.0, 2.0.0 or 2.0.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)