Create cuda 12.1 dev trigger. (#5628)

pytorch · Sep 20, 2023 · 2fda248 · 2fda248
1 parent 9f0d625
commit 2fda248
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/infra/tpu-pytorch-releases/dev_images.auto.tfvars b/infra/tpu-pytorch-releases/dev_images.auto.tfvars
@@ -8,4 +8,9 @@ dev_images = [
     cuda_version = "11.8"
     extra_tags   = ["cuda"]
   }
+  {
+    accelerator  = "cuda"
+    cuda_version = "12.1"
+    extra_tags   = ["cuda"]
+  }
 ]