diff --git a/nnunet_for_pytorch.def b/nnunet_for_pytorch.def
index 185e71cd72998b520093159f64268ffaec422c7f..ecdd68ce65e0bc51851933c5003969b50603c4c1 100644
--- a/nnunet_for_pytorch.def
+++ b/nnunet_for_pytorch.def
@@ -1,2 +1,25 @@
 Bootstrap: localimage
 From: /proj/nsc_testing/xuan/containers/cuda_11.1.1-cudnn8-runtime-ubuntu18.04.sif
+
+
+%environment
+
+export PYTHONNOUSERSITE=True
+
+
+%post 
+
+mkdir /workspace/nnunet_pyt 
+cd /workspace/nnunet_pyt
+VERSION=21.11.0
+wget --content-disposition https://api.ngc.nvidia.com/v2/resources/nvidia/nnunet_for_pytorch/versions/$VERSION/zip -O nnunet_for_pytorch_$VERSION.zip
+unzip nnunet_for_pytorch_21.11.0.zip
+
+pip install --disable-pip-version-check -r requirements.txt
+pip install monai==0.8.0 --no-dependencies
+pip uninstall -y torchtext
+
+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
\ No newline at end of file