Downloading the vertex AI docker directly and running it locally as docker run will trigger `exit 1` error. The quick solution is to use
gcloud ai custom-jobs local-run
The detail at https://cloud.google.com/vertex-ai/docs/training/containerize-run-code-local