diff --git a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/build.txt b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/build.txt
index 2bcba44942d06aded307addf68739bd5a716e829..d7f9057085fe7d6e6215d8bbbb1098e9c38d23d0 100644
--- a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/build.txt
+++ b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/build.txt
@@ -10,5 +10,14 @@ DEF_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet/
 apptainer build $CONTAINER_DIR $DEF_DIR
 
 
+
+MODULE_NAME=nnunet_for_pytorch
+MODULE_VERSION=latest
+CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}_${MODULE_VERSION}.sif
+DEF_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet/${MODULE_NAME}_${MODULE_VERSION}.def
+
+apptainer build $CONTAINER_DIR $DEF_DIR
+
+
 # Known issue 
 https://github.com/NVIDIA/DeepLearningExamples/issues/1113
diff --git a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_latest.def b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_latest.def
new file mode 100644
index 0000000000000000000000000000000000000000..fabd296bf34d493813b7f20947dd70d3a994b533
--- /dev/null
+++ b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_latest.def
@@ -0,0 +1,29 @@
+Bootstrap: localimage
+From: /proj/nsc_testing/xuan/containers/nvidia_pytorch_21.11-py3.sif
+
+
+%environment
+
+export PYTHONNOUSERSITE=True
+export OMP_NUM_THREADS=2
+
+%post 
+
+VERSION=latest
+
+mkdir /workspace/nnunet_pyt 
+cd /workspace/nnunet_pyt
+git clone https://github.com/NVIDIA/DeepLearningExamples.git 
+mv DeepLearningExamples/PyTorch/Segmentation/nnUNet ./
+rm -rf DeepLearningExamples
+
+pip install --disable-pip-version-check -r requirements.txt
+pip install monai==1.0.0 --no-dependencies
+pip pip install numpy --upgrade
+
+curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip"
+unzip -qq awscliv2.zip
+./aws/install
+rm -rf awscliv2.zip aws
+
+cp utils/instance_norm.py /usr/local/lib/python3.8/dist-packages/apex/normalization