diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars index f958d2e4ac88..13745551f828 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.