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

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

Error:
Kernel died

Open sagemaker-debugger/pytorch_profiling/pt-resnet-profiling-multi-gpu-multi-node.ipynb and begin by reviewing or running the CI workflow that reports "Kernel died." Investigate the notebook failure in its multi-GPU, multi-node context; done means the notebook completes CI without the kernel dying.

Written by the indexing model from the issue text.