nightlies with pytorch 1.7.1

Summary: Add nightly linux conda builds with the new pytorch 1.7.1. This supports python 3.9.

Reviewed By: nikhilaravi

Differential Revision: D25532757

fbshipit-source-id: b734637063e148389951899450566275f3cf5831

.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

.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,