pt-resnet-profiling-single-gpu-single-node.ipynb failed CI

Link to the notebook:
https://github.com/aws/amazon-sagemaker-examples/blob/master/sagemaker-debugger/pytorch_profiling/pt-resnet-profiling-single-gpu-single-node.ipynb

Error:
Kernel died

Start with sagemaker-debugger/pytorch_profiling/pt-resnet-profiling-single-gpu-single-node.ipynb and reproduce the CI failure. Inspect the notebook execution output and CI configuration to determine why the kernel dies; done means the notebook completes successfully in CI.

Written by the indexing model from the issue text.