diff --git a/.circleci/config.yml b/.circleci/config.yml index ce85d81e..f9b6630e 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -308,6 +308,24 @@ workflows: name: linux_conda_py36_cu110_pyt170 python_version: '3.6' pytorch_version: 1.7.0 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu101 + name: linux_conda_py36_cu101_pyt171 + python_version: '3.6' + pytorch_version: 1.7.1 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu102 + name: linux_conda_py36_cu102_pyt171 + python_version: '3.6' + pytorch_version: 1.7.1 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu110 + name: linux_conda_py36_cu110_pyt171 + python_version: '3.6' + pytorch_version: 1.7.1 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu92 @@ -392,6 +410,24 @@ workflows: name: linux_conda_py37_cu110_pyt170 python_version: '3.7' pytorch_version: 1.7.0 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu101 + name: linux_conda_py37_cu101_pyt171 + python_version: '3.7' + pytorch_version: 1.7.1 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu102 + name: linux_conda_py37_cu102_pyt171 + python_version: '3.7' + pytorch_version: 1.7.1 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu110 + name: linux_conda_py37_cu110_pyt171 + python_version: '3.7' + pytorch_version: 1.7.1 - binary_linux_conda: context: DOCKERHUB_TOKEN cu_version: cu92 @@ -476,6 +512,42 @@ workflows: name: linux_conda_py38_cu110_pyt170 python_version: '3.8' pytorch_version: 1.7.0 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu101 + name: linux_conda_py38_cu101_pyt171 + python_version: '3.8' + pytorch_version: 1.7.1 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu102 + name: linux_conda_py38_cu102_pyt171 + python_version: '3.8' + pytorch_version: 1.7.1 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu110 + name: linux_conda_py38_cu110_pyt171 + python_version: '3.8' + pytorch_version: 1.7.1 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu101 + name: linux_conda_py39_cu101_pyt171 + python_version: '3.9' + pytorch_version: 1.7.1 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu102 + name: linux_conda_py39_cu102_pyt171 + python_version: '3.9' + pytorch_version: 1.7.1 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu110 + name: linux_conda_py39_cu110_pyt171 + python_version: '3.9' + pytorch_version: 1.7.1 - binary_linux_wheel: context: DOCKERHUB_TOKEN cu_version: cu101 diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index 46be50c1..fbf5f51a 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -20,14 +20,22 @@ CONDA_CUDA_VERSIONS = { "1.5.1": ["cu92", "cu101", "cu102"], "1.6.0": ["cu92", "cu101", "cu102"], "1.7.0": ["cu101", "cu102", "cu110"], + "1.7.1": ["cu101", "cu102", "cu110"], } +def pytorch_versions_for_python(python_version): + if python_version in ["3.6", "3.7", "3.8"]: + return list(CONDA_CUDA_VERSIONS) + pytorch_without_py39 = ["1.4", "1.5.0", "1.5.1", "1.6.0", "1.7.0"] + return [i for i in CONDA_CUDA_VERSIONS if i not in pytorch_without_py39] + + def workflows(prefix="", filter_branch=None, upload=False, indentation=6): w = [] for btype in ["conda"]: - for python_version in ["3.6", "3.7", "3.8"]: - for pytorch_version in ["1.4", "1.5.0", "1.5.1", "1.6.0", "1.7.0"]: + for python_version in ["3.6", "3.7", "3.8", "3.9"]: + for pytorch_version in pytorch_versions_for_python(python_version): for cu_version in CONDA_CUDA_VERSIONS[pytorch_version]: w += workflow_pair( btype=btype,