Update 2 files

- /PyTorch/Segmentation/nnUNet/build.txt
- /PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_21.11.0.def

PyTorch/Segmentation/nnUNet/build.txt

@@ -29,6 +29,7 @@ apptainer exec --nv --pwd /workspace/nnunet_pyt $CONTAINER_DIR  python /workspac
  
 
 apptainer exec --nv --pwd /workspace/nnunet_pyt $CONTAINER_DIR python scripts/benchmark.py --mode train --gpus 1 --dim 2 --batch_size 256 --amp
+apptainer exec --nv --pwd /workspace/nnunet_pyt $CONTAINER_DIR python scripts/benchmark.py --mode predict --gpus 1 --dim 2 --batch_size 256 --amp
 
 
 # Known issue 

PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_21.11.0.def

@@ -17,10 +17,20 @@ wget --content-disposition https://api.ngc.nvidia.com/v2/resources/nvidia/nnunet
 unzip nnunet_for_pytorch_21.11.0.zip
 
 # Fix for https://github.com/NVIDIA/DeepLearningExamples/issues/1113
-sed 's/pytorch-lightning==1.3.8/pytorch-lightning==1.5.10/g' requirements.txt > temp_file && mv temp_file requirements.txt
-pip install --disable-pip-version-check -r requirements.txt
-pip install monai==0.8.0 --no-dependencies
-pip uninstall -y torchtext
+#sed 's/pytorch-lightning==1.3.8/pytorch-lightning==1.5.10/g' requirements.txt > temp_file && mv temp_file requirements.txt
+#pip install --disable-pip-version-check -r requirements.txt
+#pip install monai==0.8.0 --no-dependencies
+#pip uninstall -y torchtext
+
+conda install -y nibabel==3.2.1 \
+joblib==1.0.1 \
+pytorch-lightning==1.3.8 \
+scikit-learn==1.0 \
+scikit-image==0.18.3 \
+pynvml==11.0.0 \
+monai==0.8.0
+
+conda remove -y torchtext
 
 curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip"
 unzip -qq awscliv2.zip