conda pytorch2.0.0 builds (#1480)

Summary:
Pull Request resolved: https://github.com/facebookresearch/pytorch3d/pull/1480

Nightly builds with PyTorch 2.0.

Reviewed By: shapovalov

Differential Revision: D44135997

fbshipit-source-id: 12b363e2eadbda7a9b6ba9d8db376f41b96d551c

.circleci/config.yml

@@ -355,6 +355,20 @@ workflows:
           name: linux_conda_py38_cu117_pyt1131
           python_version: '3.8'
           pytorch_version: 1.13.1
+      - binary_linux_conda:
+          conda_docker_image: pytorch/conda-builder:cuda117
+          context: DOCKERHUB_TOKEN
+          cu_version: cu117
+          name: linux_conda_py38_cu117_pyt200
+          python_version: '3.8'
+          pytorch_version: 2.0.0
+      - binary_linux_conda:
+          conda_docker_image: pytorch/conda-builder:cuda118
+          context: DOCKERHUB_TOKEN
+          cu_version: cu118
+          name: linux_conda_py38_cu118_pyt200
+          python_version: '3.8'
+          pytorch_version: 2.0.0
       - binary_linux_conda:
           context: DOCKERHUB_TOKEN
           cu_version: cu102
@@ -530,6 +544,20 @@ workflows:
           name: linux_conda_py39_cu117_pyt1131
           python_version: '3.9'
           pytorch_version: 1.13.1
+      - binary_linux_conda:
+          conda_docker_image: pytorch/conda-builder:cuda117
+          context: DOCKERHUB_TOKEN
+          cu_version: cu117
+          name: linux_conda_py39_cu117_pyt200
+          python_version: '3.9'
+          pytorch_version: 2.0.0
+      - binary_linux_conda:
+          conda_docker_image: pytorch/conda-builder:cuda118
+          context: DOCKERHUB_TOKEN
+          cu_version: cu118
+          name: linux_conda_py39_cu118_pyt200
+          python_version: '3.9'
+          pytorch_version: 2.0.0
       - binary_linux_conda:
           context: DOCKERHUB_TOKEN
           cu_version: cu102
@@ -624,6 +652,20 @@ workflows:
           name: linux_conda_py310_cu117_pyt1131
           python_version: '3.10'
           pytorch_version: 1.13.1
+      - binary_linux_conda:
+          conda_docker_image: pytorch/conda-builder:cuda117
+          context: DOCKERHUB_TOKEN
+          cu_version: cu117
+          name: linux_conda_py310_cu117_pyt200
+          python_version: '3.10'
+          pytorch_version: 2.0.0
+      - binary_linux_conda:
+          conda_docker_image: pytorch/conda-builder:cuda118
+          context: DOCKERHUB_TOKEN
+          cu_version: cu118
+          name: linux_conda_py310_cu118_pyt200
+          python_version: '3.10'
+          pytorch_version: 2.0.0
       - binary_linux_conda_cuda:
           name: testrun_conda_cuda_py38_cu102_pyt190
           context: DOCKERHUB_TOKEN

.circleci/regenerate.py

@@ -30,32 +30,21 @@ CONDA_CUDA_VERSIONS = {
     "1.12.1": ["cu102", "cu113", "cu116"],
     "1.13.0": ["cu116", "cu117"],
     "1.13.1": ["cu116", "cu117"],
+    "2.0.0": ["cu117", "cu118"],
 }
 
 
 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"
-    if cuda_version == "cu116":
-        return "pytorch/conda-builder:cuda116"
-    if cuda_version == "cu117":
-        return "pytorch/conda-builder:cuda117"
-    raise ValueError("Unknown cuda version")
+    if len(cuda_version) != 5:
+        raise ValueError("Unknown cuda version")
+    return "pytorch/conda-builder:cuda" + cuda_version[2:]
 
 
 def pytorch_versions_for_python(python_version):
-    if python_version in ["3.7", "3.8"]:
+    if python_version in ["3.8", "3.9"]:
         return list(CONDA_CUDA_VERSIONS)
-    if python_version == "3.9":
-        return [
-            i
-            for i in CONDA_CUDA_VERSIONS
-            if version.Version(i) > version.Version("1.7.0")
-        ]
     if python_version == "3.10":
         return [
             i