Addition of -cpuonly flag for Vertex AI training

GoogleCloudPlatform / data-science-on-gcp

Source code accompanying book: Data Science on the Google Cloud Platform, Valliappa Lakshmanan, O'Reilly 2017

Apache License 2.0

1.31k stars 715 forks source link

Addition of -cpuonly flag for Vertex AI training #144

Closed joemansfield closed 2 years ago

joemansfield commented 2 years ago

Added command line switch --cpuonly to 10_mlops/train_on_vertexai.py example in Chapter 10 to select cpu images for situations where GPU accelerators are not available or undesirable.

joemansfield commented 2 years ago

@lakshmanok This is the updated train_on_vertexai.pyscipt with the --cpuonlyoption that allows us to run this in Qwiklabs. Can you review and merge?