From 508ac52fda3ca53ac3261e8465b455facfd5e2fb Mon Sep 17 00:00:00 2001 From: iefgnoix Date: Thu, 21 Mar 2024 03:16:09 +0000 Subject: [PATCH] Add cuda 12.1 nightlies for cp310 and cp311. --- infra/tpu-pytorch-releases/artifacts.auto.tfvars | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars index f958d2e4ac8..13745551f82 100644 --- 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.