Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
An example of how to launch a training ZenML pipeline on a GPU accelerated Google Cloud VM
training_pipeline.run(
backend=OrchestratorGCPBackend(
preemptible=True, # reduce costs by using preemptible instance
machine_type='n1-standard-4',
gpu='nvidia-tesla-k80',
gpu_count=1,
)
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment