AWS Certified Machine Learning – Specialty — Question 25

A Machine Learning Specialist is packaging a custom ResNet model into a Docker container so the company can leverage Amazon SageMaker for training. The
Specialist is using Amazon EC2 P3 instances to train the model and needs to properly configure the Docker container to leverage the NVIDIA GPUs.
What does the Specialist need to do?

Answer options

• A. Bundle the NVIDIA drivers with the Docker image.
• B. Build the Docker container to be NVIDIA-Docker compatible.
• C. Organize the Docker container's file structure to execute on GPU instances.
• D. Set the GPU flag in the Amazon SageMaker CreateTrainingJob request body.

Correct answer: B

Explanation

The correct answer is B because building the Docker container to be NVIDIA-Docker compatible ensures that it can effectively use the NVIDIA GPU resources available on the EC2 P3 instances. Option A is incorrect because bundling NVIDIA drivers is not necessary when using NVIDIA-Docker, which manages this for you. Option C is irrelevant as the structure alone does not guarantee GPU utilization, and Option D is incorrect since setting a flag does not directly configure the container to use NVIDIA GPUs.