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?
- Bundle the NVIDIA drivers with the Docker image.
- Build the Docker container to be NVIDIA-Docker compatible.
- Organize the Docker container's file structure to execute on GPU instances.
- Set the GPU flag in the Amazon SageMaker CreateTrainingJob request body.
Answer(s): B
Explanation:
To leverage the NVIDIA GPUs on Amazon EC2 P3 instances for training with Amazon SageMaker, the Docker container must be built to be compatible with NVIDIA-Docker.
NVIDIA-Docker is a wrapper around Docker that makes it easier to use GPUs in containers by providing GPU-aware functionality.
To build a Docker container that is compatible with NVIDIA-Docker, the Specialist should install the NVIDIA GPU drivers in the Docker container and install the NVIDIA-Docker runtime on the EC2 instances.
Reference:
https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-dg.pdf
Reveal Solution Next Question