From d0ca3b9e0cf6b1cfba46a367a98b8738cc5acad5 Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Mon, 27 Sep 2021 04:15:53 -0700 Subject: [PATCH] builds for PyTorch 1.9.1 Summary: Add conda builds for the newly released PyTorch version 1.9.1. Reviewed By: patricklabatut Differential Revision: D31140206 fbshipit-source-id: 697549a3ef0db8248f4f9b5c00cf1407296b5022 --- .circleci/config.yml | 48 ++++++++++++++++++++++++++++++++ .circleci/regenerate.py | 1 + packaging/linux_wheels/inside.sh | 1 + 3 files changed, 50 insertions(+) diff --git a/.circleci/config.yml b/.circleci/config.yml index c1f06e19..b5dd92ef 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -368,6 +368,18 @@ workflows: name: linux_conda_py36_cu111_pyt190 python_version: '3.6' pytorch_version: 1.9.0 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu102 + name: linux_conda_py36_cu102_pyt191 + python_version: '3.6' + pytorch_version: 1.9.1 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu111 + name: linux_conda_py36_cu111_pyt191 + python_version: '3.6' + pytorch_version: 1.9.1 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu92 @@ -506,6 +518,18 @@ workflows: name: linux_conda_py37_cu111_pyt190 python_version: '3.7' pytorch_version: 1.9.0 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu102 + name: linux_conda_py37_cu102_pyt191 + python_version: '3.7' + pytorch_version: 1.9.1 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu111 + name: linux_conda_py37_cu111_pyt191 + python_version: '3.7' + pytorch_version: 1.9.1 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu92 @@ -644,6 +668,18 @@ workflows: 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: cu101 @@ -710,6 +746,18 @@ workflows: 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 - binary_linux_conda_cuda: name: testrun_conda_cuda_py36_cu101_pyt151 context: DOCKERHUB_TOKEN diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index d8d617d7..9457a853 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -27,6 +27,7 @@ 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"], } diff --git a/packaging/linux_wheels/inside.sh b/packaging/linux_wheels/inside.sh index ba8a16b6..81533af1 100644 --- a/packaging/linux_wheels/inside.sh +++ b/packaging/linux_wheels/inside.sh @@ -38,6 +38,7 @@ declare -A 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" )