diff --git a/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch.def b/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch.def new file mode 100644 index 0000000000000000000000000000000000000000..f5803343dee7840aa6e710b1a652b26938dc7994 --- /dev/null +++ b/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch.def @@ -0,0 +1,26 @@ +Bootstrap: localimage +From: /proj/nsc_testing/xuan/containers/nvidia_pytorch_21.11-py3.sif + + +%environment + +export PYTHONNOUSERSITE=True + + +%post + +VERSION=21.11.0 + +mkdir /workspace/nnunet_pyt +cd /workspace/nnunet_pyt +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