runtime error

Exit code: 1. Reason: sions/3.10.12/lib/python3.10/site-packages/megatron/core/tensor_parallel/layers.py:254: FutureWarning: `torch.cuda.amp.custom_fwd(args...)` is deprecated. Please use `torch.amp.custom_fwd(args..., device_type='cuda')` instead. def forward( [NeMo W 2025-11-10 00:19:56 nemo_logging:349] /home/user/.pyenv/versions/3.10.12/lib/python3.10/site-packages/megatron/core/tensor_parallel/layers.py:265: FutureWarning: `torch.cuda.amp.custom_bwd(args...)` is deprecated. Please use `torch.amp.custom_bwd(args..., device_type='cuda')` instead. def backward(ctx, grad_output): [NeMo W 2025-11-10 00:19:56 nemo_logging:349] /home/user/.pyenv/versions/3.10.12/lib/python3.10/site-packages/megatron/core/tensor_parallel/layers.py:325: FutureWarning: `torch.cuda.amp.custom_fwd(args...)` is deprecated. Please use `torch.amp.custom_fwd(args..., device_type='cuda')` instead. def forward( [NeMo W 2025-11-10 00:19:56 nemo_logging:349] /home/user/.pyenv/versions/3.10.12/lib/python3.10/site-packages/megatron/core/tensor_parallel/layers.py:360: FutureWarning: `torch.cuda.amp.custom_bwd(args...)` is deprecated. Please use `torch.amp.custom_bwd(args..., device_type='cuda')` instead. def backward(ctx, grad_output): Traceback (most recent call last): File "/home/user/app/app.py", line 7, in <module> from nemo.collections.nlp.models import PunctuationCapitalizationModel File "/home/user/.pyenv/versions/3.10.12/lib/python3.10/site-packages/nemo/collections/nlp/__init__.py", line 15, in <module> from nemo.collections.nlp import data, losses, models, modules File "/home/user/.pyenv/versions/3.10.12/lib/python3.10/site-packages/nemo/collections/nlp/data/__init__.py", line 33, in <module> from nemo.collections.nlp.data.question_answering_squad.qa_dataset import SquadDataset File "/home/user/.pyenv/versions/3.10.12/lib/python3.10/site-packages/nemo/collections/nlp/data/question_answering_squad/qa_dataset.py", line 25, in <module> import psutil ModuleNotFoundError: No module named 'psutil'

Container logs: