The answer is to use a custom machine type with more memory, as the immediate cause of an 'Out of memory: Killed process' error during TensorFlow training on Vertex AI is system-wide memory exhaustion. While enabling memory growth with `tf.config.experimental.set_memory_growth` prevents TensorFlow from greedily allocating all GPU memory, the core issue here is that the total available RAM on the default machine is insufficient for both the TensorFlow process and the operating system, leading to the kernel killing the job. On the Google Professional Data Engineer exam, this scenario tests your ability to diagnose resource constraints in Vertex AI pipelines, where a common trap is to only focus on GPU memory settings while ignoring the underlying machine's RAM limits. A key memory tip: when you see "Killed process" in Vertex AI, think "RAM starvation first"—always scale up the machine type before tuning TensorFlow's memory allocation.
PDE Operationalizing machine learning models Practice Question
This PDE practice question tests your understanding of operationalizing machine learning models. The scenario asks you to isolate a root cause — eliminate options that address a different problem before choosing. After answering, compare your reasoning against the explanation and wrong-answer breakdown below. Once you have made your selection, read the full explanation to reinforce the concept and understand why each distractor is designed to mislead on exam day.
During a Vertex AI training pipeline, the training job fails with an error: 'Out of memory: Killed process'. The model is a large deep learning model using TensorFlow. Which THREE steps should the team take to resolve this issue?
Answer the question above first, then reveal the full breakdown to understand why each option is right or wrong.
Correct answer & explanation
✓
Enable memory growth configuration in TensorFlow
Option B is correct because TensorFlow by default allocates all available GPU memory, which can cause out-of-memory (OOM) errors when other processes or the system itself need memory. Enabling memory growth with `tf.config.experimental.set_memory_growth` allows TensorFlow to allocate memory incrementally, reducing the risk of OOM kills. This is a direct mitigation for the 'Killed process' error caused by memory exhaustion.
Key principle: Answer the scenario, not the keyword: identify the specific constraint before choosing the most familiar-sounding option.
Answer analysis
Option-by-option breakdown
For each option: why learners choose it and why it is or isn't the right answer here.
✗
Change to a distributed training strategy
Why it's wrong here
Distributed training can help with large models but is not a direct fix for OOM; it requires code changes.
✓
Enable memory growth configuration in TensorFlow
Why this is correct
Memory growth allows TensorFlow to allocate memory on demand, avoiding early OOM.
Related concept
Read the scenario before looking for a memorised answer.
✗
Switch the training from GPU to TPU accelerator
Why it's wrong here
TPUs have different memory characteristics; this is not a straightforward fix for OOM.
✓
Reduce the training batch size
Why this is correct
Smaller batch sizes reduce memory footprint.
Related concept
Read the scenario before looking for a memorised answer.
✓
Use a custom machine type with more memory
Why this is correct
Increasing memory capacity prevents OOM.
Related concept
Read the scenario before looking for a memorised answer.
Common exam traps
Common exam trap: answer the scenario, not the keyword
Google Cloud often tests the misconception that distributed training automatically solves memory issues, but in reality, it distributes computation, not memory pressure, and can even increase per-node memory usage due to gradient synchronization buffers.
Detailed technical explanation
How to think about this question
Under the hood, TensorFlow's default behavior calls `cudaMalloc` to pre-allocate nearly all GPU memory, which can starve other processes and trigger the OOM killer. Enabling memory growth uses `cudaMallocAsync` or incremental allocation, allowing the GPU memory to be shared with other workloads. In Vertex AI, the training container runs on a Compute Engine VM; if the VM's RAM is insufficient for both the TensorFlow process and system overhead, the kernel's OOM killer terminates the process, which is why increasing VM memory (Option E) is also effective.
KKey Concepts to Remember
Read the scenario before looking for a memorised answer.
Find the constraint that changes the correct option.
Eliminate answers that are true in general but not in this case.
TExam Day Tips
→Watch for words such as best, first, most likely and least administrative effort.
→Review why wrong options are wrong, not only why the correct option is correct.
Key takeaway
Answer the scenario, not the keyword: identify the specific constraint before choosing the most familiar-sounding option.
Real-world example
How this comes up in practice
A cloud solutions architect for a retail company is evaluating services for a new workload. The correct answer here reflects best practice for the specific scenario described — not a general cloud recommendation. Answer the scenario, not the keyword: identify the specific constraint before choosing the most familiar-sounding option. Cloud exam questions reward reading the constraint carefully: the same technology can be right or wrong depending on the use case.
What to study next
Got this wrong? Here's your next step.
Identify which exam domain this question belongs to, review the core concept, then practise similar questions from the same domain.
Operationalizing machine learning models — This question tests Operationalizing machine learning models — Read the scenario before looking for a memorised answer..
What is the correct answer to this question?
The correct answer is: Enable memory growth configuration in TensorFlow — Option B is correct because TensorFlow by default allocates all available GPU memory, which can cause out-of-memory (OOM) errors when other processes or the system itself need memory. Enabling memory growth with `tf.config.experimental.set_memory_growth` allows TensorFlow to allocate memory incrementally, reducing the risk of OOM kills. This is a direct mitigation for the 'Killed process' error caused by memory exhaustion.
What should I do if I get this PDE question wrong?
Identify which exam domain this question belongs to, review the core concept, then practise similar questions from the same domain.
What is the key concept behind this question?
Read the scenario before looking for a memorised answer.
About these practice questions
Courseiva creates original exam-style practice questions with explanations and wrong-answer analysis. It does not publish real exam questions, exam dumps, or protected exam content. Learn why practice questions differ from exam dumps →
Share a tip, memory trick, or ask about the reasoning behind this question. Do not post real exam questions, leaked content, braindumps, or copyrighted exam material. Comments are moderated and may be removed without notice.
This PDE practice question is part of Courseiva's free Google Cloud certification practice question bank. Courseiva provides original exam-style practice questions with explanations, topic-based practice, mock exams, readiness tracking, and study analytics to help learners prepare for the PDE exam.
Question Discussion
Share a tip, memory trick, or ask about the reasoning behind this question. Do not post real exam questions, leaked content, braindumps, or copyrighted exam material. Comments are moderated and may be removed without notice.
Sign in to join the discussion.