cu116 docker image

Summary: cu116 builds need to happen in a specific image.

Reviewed By: patricklabatut

Differential Revision: D37680352

fbshipit-source-id: 81bef0642ad832e83e4eba6321287759b3229303

.circleci/config.yml

@@ -337,6 +337,7 @@ workflows:
           python_version: '3.7'
           pytorch_version: 1.12.0
       - binary_linux_conda:
+          conda_docker_image: pytorch/conda-builder:cuda116
           context: DOCKERHUB_TOKEN
           cu_version: cu116
           name: linux_conda_py37_cu116_pyt1120
@@ -499,6 +500,7 @@ workflows:
           python_version: '3.8'
           pytorch_version: 1.12.0
       - binary_linux_conda:
+          conda_docker_image: pytorch/conda-builder:cuda116
           context: DOCKERHUB_TOKEN
           cu_version: cu116
           name: linux_conda_py38_cu116_pyt1120
@@ -661,6 +663,7 @@ workflows:
           python_version: '3.9'
           pytorch_version: 1.12.0
       - binary_linux_conda:
+          conda_docker_image: pytorch/conda-builder:cuda116
           context: DOCKERHUB_TOKEN
           cu_version: cu116
           name: linux_conda_py39_cu116_pyt1120
@@ -706,6 +709,7 @@ workflows:
           python_version: '3.10'
           pytorch_version: 1.12.0
       - binary_linux_conda:
+          conda_docker_image: pytorch/conda-builder:cuda116
           context: DOCKERHUB_TOKEN
           cu_version: cu116
           name: linux_conda_py310_cu116_pyt1120

.circleci/regenerate.py

@@ -33,11 +33,15 @@ CONDA_CUDA_VERSIONS = {
 
 
 def conda_docker_image_for_cuda(cuda_version):
+    if cuda_version in ("cu101", "cu102", "cu111"):
+        return None
     if cuda_version == "cu113":
         return "pytorch/conda-builder:cuda113"
     if cuda_version == "cu115":
         return "pytorch/conda-builder:cuda115"
-    return None
+    if cuda_version == "cu116":
+        return "pytorch/conda-builder:cuda116"
+    raise ValueError("Unknown cuda version")
 
 
 def pytorch_versions_for_python(python_version):

packaging/linux_wheels/go.sh

@@ -8,3 +8,4 @@
 sudo docker run --rm  -v "$PWD/../../:/inside" pytorch/conda-cuda bash inside/packaging/linux_wheels/inside.sh
 sudo docker run --rm  -v "$PWD/../../:/inside" -e SELECTED_CUDA=cu113 pytorch/conda-builder:cuda113 bash inside/packaging/linux_wheels/inside.sh
 sudo docker run --rm  -v "$PWD/../../:/inside" -e SELECTED_CUDA=cu115 pytorch/conda-builder:cuda115 bash inside/packaging/linux_wheels/inside.sh
+sudo docker run --rm  -v "$PWD/../../:/inside" -e SELECTED_CUDA=cu116 pytorch/conda-builder:cuda116 bash inside/packaging/linux_wheels/inside.sh