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