diff --git a/README.md b/README.md index 1f7ef07f97cb833a27501ac9767a456b16524c48..e58602f636363ed5e578f739cb254ee18ac4e9ec 100644 --- a/README.md +++ b/README.md @@ -4,14 +4,14 @@ Latest Version 21.11.0 Modified February 3, 2022 See [NVIDIA Deep Learning Examples](https://catalog.ngc.nvidia.com/orgs/nvidia/resources/nnunet_for_pytorch). -### On local -- Clone the repository. +### On local computer +- Clone the repository ``` git clone https://github.com/NVIDIA/DeepLearningExamples cd ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet ``` -- Build the nnU-Net PyTorch NGC container. +- Build the nnU-Net PyTorch NGC container ``` docker build -t nnunet . ``` @@ -24,14 +24,31 @@ docker push xuagu37/nvidia_nnu-net_for_pytorch:21.11.0 ### On Berzelius -- Clone the repository. +- Clone the repository ``` +cd /proj/nsc/xuan/ngc git clone https://github.com/NVIDIA/DeepLearningExamples -cd ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet ``` -- On Berzelius, pull the image for Singularity and run +#### For single node +- Start an interactive session +``` +interactive -N2 --reservation=nsc-testing -t 600 ``` + +- Pull the image for Singularity and run +``` +cd /proj/nsc/xuan/ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet singularity pull nvidia_nnu-net_for_pytorch.sif docker://xuagu37/nvidia_nnu-net_for_pytorch:21.11.0 singularity shell -B ${PWD}/data:/data -B ${PWD}/results:/results --nv nvidia_nnu-net_for_pytorch.sif ``` +- Run the benchmark script +``` +bash benchmark_nnunet_pytorch_berzelius.sh +``` + +#### For multi-node +- Run the benchmark script +``` +sbash benchmark_nnunet_pytorch_berzelius_multi_node.sh +```