From b6771233e0f546da55fb4ea31799c1a1637a7d4a Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Mon, 15 Aug 2022 08:51:12 -0700 Subject: [PATCH] builds for PyTorch 1.12.1, drop 1.8 Summary: Builds for new PyTorch 1.12.1. Drop builds for PyTorch 1.8.0 and 1.8.1. Reviewed By: kjchalup Differential Revision: D38658991 fbshipit-source-id: 6192e226c2154cd051eeee98498d9a395cfd6fd5 --- .circleci/config.yml | 160 +++++++++++++++++----------------------- .circleci/regenerate.py | 3 +- 2 files changed, 67 insertions(+), 96 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 45c98665..585f09df 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -180,42 +180,6 @@ workflows: jobs: # - main: # context: DOCKERHUB_TOKEN - - 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 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu102 @@ -343,42 +307,26 @@ workflows: name: linux_conda_py37_cu116_pyt1120 python_version: '3.7' pytorch_version: 1.12.0 - - 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 + name: linux_conda_py37_cu102_pyt1121 + python_version: '3.7' + pytorch_version: 1.12.1 - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda113 context: DOCKERHUB_TOKEN - cu_version: cu111 - name: linux_conda_py38_cu111_pyt180 - python_version: '3.8' - pytorch_version: 1.8.0 + cu_version: cu113 + name: linux_conda_py37_cu113_pyt1121 + python_version: '3.7' + pytorch_version: 1.12.1 - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 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 + cu_version: cu116 + name: linux_conda_py37_cu116_pyt1121 + python_version: '3.7' + pytorch_version: 1.12.1 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu102 @@ -506,42 +454,26 @@ workflows: name: linux_conda_py38_cu116_pyt1120 python_version: '3.8' pytorch_version: 1.12.0 - - 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 + name: linux_conda_py38_cu102_pyt1121 + python_version: '3.8' + pytorch_version: 1.12.1 - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda113 context: DOCKERHUB_TOKEN - cu_version: cu111 - name: linux_conda_py39_cu111_pyt180 - python_version: '3.9' - pytorch_version: 1.8.0 + cu_version: cu113 + name: linux_conda_py38_cu113_pyt1121 + python_version: '3.8' + pytorch_version: 1.12.1 - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 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 + cu_version: cu116 + name: linux_conda_py38_cu116_pyt1121 + python_version: '3.8' + pytorch_version: 1.12.1 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu102 @@ -669,6 +601,26 @@ workflows: name: linux_conda_py39_cu116_pyt1120 python_version: '3.9' pytorch_version: 1.12.0 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu102 + name: linux_conda_py39_cu102_pyt1121 + python_version: '3.9' + pytorch_version: 1.12.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda113 + context: DOCKERHUB_TOKEN + cu_version: cu113 + name: linux_conda_py39_cu113_pyt1121 + python_version: '3.9' + pytorch_version: 1.12.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 + context: DOCKERHUB_TOKEN + cu_version: cu116 + name: linux_conda_py39_cu116_pyt1121 + python_version: '3.9' + pytorch_version: 1.12.1 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu102 @@ -715,6 +667,26 @@ workflows: name: linux_conda_py310_cu116_pyt1120 python_version: '3.10' pytorch_version: 1.12.0 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu102 + name: linux_conda_py310_cu102_pyt1121 + python_version: '3.10' + pytorch_version: 1.12.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda113 + context: DOCKERHUB_TOKEN + cu_version: cu113 + name: linux_conda_py310_cu113_pyt1121 + python_version: '3.10' + pytorch_version: 1.12.1 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda116 + context: DOCKERHUB_TOKEN + cu_version: cu116 + name: linux_conda_py310_cu116_pyt1121 + python_version: '3.10' + pytorch_version: 1.12.1 - binary_linux_conda_cuda: name: testrun_conda_cuda_py37_cu102_pyt190 context: DOCKERHUB_TOKEN diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index 6892afa4..8fa8a446 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.8.0": ["cu101", "cu102", "cu111"], - "1.8.1": ["cu101", "cu102", "cu111"], "1.9.0": ["cu102", "cu111"], "1.9.1": ["cu102", "cu111"], "1.10.0": ["cu102", "cu111", "cu113"], @@ -29,6 +27,7 @@ CONDA_CUDA_VERSIONS = { "1.10.2": ["cu102", "cu111", "cu113"], "1.11.0": ["cu102", "cu111", "cu113", "cu115"], "1.12.0": ["cu102", "cu113", "cu116"], + "1.12.1": ["cu102", "cu113", "cu116"], }