update A100 to use us-central1-f (#100)

GoogleCloudPlatform · Feb 5, 2024 · 693ee3f · 693ee3f
1 parent 9d89ba0
commit 693ee3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dags/pytorch_xla/pytorchxla_torchbench.py b/dags/pytorch_xla/pytorchxla_torchbench.py
@@ -93,7 +93,7 @@
       image_family=resource.ImageFamily.COMMON_CU121_DEBIAN_11,
       accelerator_type=resource.GpuVersion.A100,
       count=1,
-      gpu_zone=resource.Zone.US_CENTRAL1_C,
+      gpu_zone=resource.Zone.US_CENTRAL1_F,
       model_name=model,
       time_out_in_min=1600,
       extraFlags=" ".join(torchbench_extra_flags),