Update the Dockerfile with necessary changes from NeMo for H100 self attention accuracy.

Update the Dockerfile with necessary changes from NeMo for H100 self attention accuracy. #2829