Skip to content

Instantly share code, notes, and snippets.

@htahir1

htahir1/zenml_gcp_gpu_orchestration.py

Last active January 29, 2021 09:16
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save htahir1/62dc4baa12560e8b88ce156f76aaab5f to your computer and use it in GitHub Desktop.
Save htahir1/62dc4baa12560e8b88ce156f76aaab5f to your computer and use it in GitHub Desktop.
Download ZIP
An example of how to launch a training ZenML pipeline on a GPU accelerated Google Cloud VM
Raw
zenml_gcp_gpu_orchestration.py
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