Skip to content
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.

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

Add option to setup preemptible VMs #408

Merged
merged 2 commits into from
Sep 13, 2023
Merged

Conversation

PhilippeMoussalli
Copy link
Contributor

PR that adds an option to set a preemptible VM on the component spec level.
This can be done by setting preemptible to True at the Op level. Note that you also need to assign a preemptible nodepool. I have already setup some preemptible nodepools in this PR

Caveat: seems like kfp currently only enables setting up preemptibles on a GCP cluster since this functionaliteis seems to be tightly integrated with GKE. More info [here](
Preemptible nodepools can only run up to 24 hours link

We can think later on incorporating a retry mechanism once the issue with Dask is fixed. For now those VMs can be used to reduce development costs.

Copy link
Collaborator

@GeorgesLorre GeorgesLorre left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

will be very helpful for future bigger runs

Copy link
Contributor

@mrchtr mrchtr left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@PhilippeMoussalli PhilippeMoussalli merged commit 7ad42aa into main Sep 13, 2023
5 checks passed
@PhilippeMoussalli PhilippeMoussalli deleted the add-preemptible-option branch September 13, 2023 07:00
Hakimovich99 pushed a commit that referenced this pull request Oct 16, 2023
PR that adds an option to set a preemptible VM on the component spec
level.
This can be done by setting `preemptible` to True at the `Op` level.
Note that you also need to assign a preemptible nodepool. I have already
setup some preemptible nodepools in [this
PR](https://github.com/ml6team/Express-infra/compare/add-preemtibles-node-pools?expand=1)

Caveat: seems like kfp currently only enables setting up preemptibles on
a GCP cluster since this functionaliteis seems to be tightly integrated
with GKE. More info [here](
Preemptible nodepools can only run up to 24 hours
[link](https://cloud.google.com/compute/docs/instances/preemptible#:~:text=Compute%20Engine%20always%20stops%20preemptible%20instances%20after%20they%20run%20for%2024%20hours.%20Certain%20actions%20reset%20this%2024%2Dhour%20counter.)

We can think later on incorporating a retry mechanism once the issue
with Dask is fixed. For now those VMs can be used to reduce development
costs.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants