Add cuda 12.1 nightly wheels for cp310 and cp311. (#6793)

pytorch · Mar 22, 2024 · 7d8972c · 7d8972c
1 parent e75677f
commit 7d8972c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars
@@ -19,6 +19,16 @@ nightly_builds = [
     accelerator  = "cuda"
     cuda_version = "12.1"
   },
+  {
+    accelerator  = "cuda"
+    cuda_version = "12.1"
+    python_version = "3.10"
+  },
+  {
+    accelerator  = "cuda"
+    cuda_version = "12.1"
+    python_version = "3.11"
+  },
 ]
 
 # Built on push to specific tag.