Add CUDA 12.1 builds (#7533)

Windows CUDA-12.1 failures are expected
pytorch · Apr 21, 2023 · 0d75d9e · 0d75d9e
1 parent 2925df7
commit 0d75d9e
Show file tree

Hide file tree

Showing 5 changed files with 255 additions and 25 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py
@@ -32,8 +32,8 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
         for os_type in ["linux", "macos", "win"]:
             python_versions = PYTHON_VERSIONS
             cu_versions_dict = {
-                "linux": ["cpu", "cu117", "cu118", "rocm5.2", "rocm5.3"],
-                "win": ["cpu", "cu117", "cu118"],
+                "linux": ["cpu", "cu117", "cu118", "cu121", "rocm5.2", "rocm5.3"],
+                "win": ["cpu", "cu117", "cu118", "cu121"],
                 "macos": ["cpu"],
             }
             cu_versions = cu_versions_dict[os_type]
@@ -109,6 +109,7 @@ def workflow_pair(btype, os_type, python_version, cu_version, unicode, prefix=""
 manylinux_images = {
     "cu117": "pytorch/manylinux-cuda117",
     "cu118": "pytorch/manylinux-cuda118",
+    "cu121": "pytorch/manylinux-cuda121",
 }
 
 

diff --git a/.circleci/unittest/windows/scripts/install.sh b/.circleci/unittest/windows/scripts/install.sh
@@ -25,7 +25,7 @@ else
     fi
 
     cuda_toolkit_pckg="cudatoolkit"
-    if [[ $CUDA_VERSION == 11.6 || $CUDA_VERSION == 11.7 || $CUDA_VERSION == 11.8 ]]; then
+    if [[ $CUDA_VERSION == 11.6 || $CUDA_VERSION == 11.7 || $CUDA_VERSION == 11.8 || $CUDA_VERSION == 12.1 ]]; then
         cuda_toolkit_pckg="pytorch-cuda"
     fi