Update 2 files

- /NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_latest.def
- /NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/build.txt

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

NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_latest.def

+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