New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

GKE GPU A3 with TCPX support #517

Merged

markblee merged 45 commits into apple:main from samos123:gke-gpu

Jun 29, 2024

remove tpu np provionser from gpu runner

Loading branch information

samos123 committed Jun 27, 2024

commit 0db3bb81b97966f823cb7920f7bed191b6425321

axlearn/cloud/gcp/jobs/gke_runner.py

-Original file line number
+Diff line change
@@ Expand Up / @@ -450,7 +450,6 @@ class GPUGKERunnerJob(GKERunnerJob): @@
         """A GKERunnerJob that uses GPUGKEJob."""
         inner = GPUGKEJob
-        pre_provisioner = TPUNodePoolProvisioner
     def _get_runner_or_exit(instance_type: str):
@@ Expand Down @@