PyTorch 1.10 + CUDA 11.1 builds

Summary: Although the PyTorch website, which describes the current version 1.10, suggests CUDA 10.2 and 11.3 are supported, it would appear that we need to include builds for CUDA 11.1 to avoid surprises. This is because these builds are on anaconda, and this combination is used on Google Colab.

Reviewed By: nikhilaravi

Differential Revision: D33063932

fbshipit-source-id: 1b22d1f06e22bd18fb53ceecb58e78ac6a5d1693

.circleci/config.yml

@@ -340,6 +340,12 @@ workflows:
           name: linux_conda_py36_cu102_pyt1100
           python_version: '3.6'
           pytorch_version: 1.10.0
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu111
+          name: linux_conda_py36_cu111_pyt1100
+          python_version: '3.6'
+          pytorch_version: 1.10.0
       - binary_linux_conda:
           conda_docker_image: pytorch/conda-builder:cuda113
           context: DOCKERHUB_TOKEN
@@ -467,6 +473,12 @@ workflows:
           name: linux_conda_py37_cu102_pyt1100
           python_version: '3.7'
           pytorch_version: 1.10.0
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu111
+          name: linux_conda_py37_cu111_pyt1100
+          python_version: '3.7'
+          pytorch_version: 1.10.0
       - binary_linux_conda:
           conda_docker_image: pytorch/conda-builder:cuda113
           context: DOCKERHUB_TOKEN
@@ -594,6 +606,12 @@ workflows:
           name: linux_conda_py38_cu102_pyt1100
           python_version: '3.8'
           pytorch_version: 1.10.0
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu111
+          name: linux_conda_py38_cu111_pyt1100
+          python_version: '3.8'
+          pytorch_version: 1.10.0
       - binary_linux_conda:
           conda_docker_image: pytorch/conda-builder:cuda113
           context: DOCKERHUB_TOKEN
@@ -685,6 +703,12 @@ workflows:
           name: linux_conda_py39_cu102_pyt1100
           python_version: '3.9'
           pytorch_version: 1.10.0
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu111
+          name: linux_conda_py39_cu111_pyt1100
+          python_version: '3.9'
+          pytorch_version: 1.10.0
       - binary_linux_conda:
           conda_docker_image: pytorch/conda-builder:cuda113
           context: DOCKERHUB_TOKEN

.circleci/regenerate.py

@@ -26,7 +26,7 @@ CONDA_CUDA_VERSIONS = {
     "1.8.1": ["cu101", "cu102", "cu111"],
     "1.9.0": ["cu102", "cu111"],
     "1.9.1": ["cu102", "cu111"],
-    "1.10.0": ["cu102", "cu113"],
+    "1.10.0": ["cu102", "cu111", "cu113"],
 }