AI0-001 AI Implementation and Operations Practice Question
Exhibit
Refer to the exhibit. Error: RuntimeError: CUDA out of memory. Tried to allocate 2.00 GiB (GPU 0; 8.00 GiB total capacity; 6.50 GiB already allocated; 1.50 GiB free; 0 bytes cached) at /workspace/training.py:345
Based on the exhibit, which action is most likely to resolve the memory issue?
⚠ Common exam trap
CompTIA often tests the misconception that memory errors are solved by adding more data or changing hardware, when in fact the simplest and most common fix is adjusting the batch size to fit within available GPU memory.
Answer choices
Why each option matters
Answer the question above first, then reveal the full breakdown to understand why each option is right or wrong.
Correct answer & explanation
✓
Reduce the batch size.
The exhibit shows an out-of-memory (OOM) error during training. Reducing the batch size decreases the memory footprint per iteration, allowing the model to fit within available GPU memory. This directly resolves the memory issue without altering the model architecture or data.
Answer analysis
Option-by-option breakdown
For each option: why learners choose it and why it is or isn't the right answer here.
- ✗
Add more training data.
Why it's wrong here
Adding training data increases dataset size and compute load, worsening rather than relieving memory pressure during training. More data is the right remedy when the model underfits or lacks coverage, not when the exhibit shows memory exhaustion during execution.
- ✗
Increase the learning rate.
Why it's wrong here
Raising the learning rate changes gradient step size, not memory footprint; larger steps can even worsen instability without freeing a single byte. It is tempting because learning rate is the usual first lever for convergence problems, and it would be the right adjustment when training loss plateaus or diverges rather than when memory is exhausted.
- ✗
Switch to a CPU.
Why it's wrong here
Switching to CPU changes the compute device, not memory demand; the model and its tensors still occupy the same memory, so the issue persists, only slower. CPU execution suits small models or inference where GPU acceleration is unnecessary, not memory-constrained training.
- ✓
Reduce the batch size.
Why this is correct
Reducing batch size lowers peak activation memory because fewer samples are held simultaneously during the forward and backward passes. This directly addresses the memory exhaustion constraint shown in the exhibit without altering model architecture or precision.
Visual reference
About these practice questions
Courseiva writes every AI0-001 question from scratch — 962 in total, each with an explanation and a wrong-answer breakdown. None are copied from real exams or dumps. Learn why practice questions differ from exam dumps →
JA
Written by Johnson Ajibi, MSc IT Security
Senior Network & Security Engineer · founder of Courseiva
This AI0-001 practice question is part of Courseiva's free CompTIA 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 AI0-001 exam.