Add cuda 12.1 nightlies for cp310 and cp311.

pytorch · Mar 21, 2024 · 508ac52 · 508ac52
1 parent fe3f23c
commit 508ac52
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.