From d7ae4c18e3fc88e06e3bd8504fcdfbe145a747a0 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuagu37@gmail.com>
Date: Tue, 1 Nov 2022 19:45:35 +0100
Subject: [PATCH] Update README.md

---
 README.md | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 182002b..3ea6c67 100644
--- a/README.md
+++ b/README.md
@@ -33,18 +33,25 @@ cd /proj/nsc/xuan/ngc
 git clone https://github.com/NVIDIA/DeepLearningExamples
 cd /proj/nsc/xuan/ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet
 ```
-- Pull from xuagu37 and run the image
+- Pull from xuagu37 and run the image  
+Using singularity  
 ```
 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 
+```  
+Or using enroot  
 ```
-- Prepare BraTS dataset
+enroot import 'docker://xuagu37/nvidia_nnu-net_for_pytorch:21.11.0'
+enroot create --name nnunet xuagu37+nvidia_nnu-net_for_pytorch+21.11.0.sqsh
+enroot start --rw --mount ${PWD}/data:/data --mount ${PWD}/results:/results nnunet
+```
+- Prepare BraTS dataset (within the image)
 ```
 python download.py --task 01
 python preprocess.py --task 01 --dim 2
 ```
 - Run the script.   
-You need to modify the script for e.g. the name of your reservation, number of nodes, batch_size, etc.
+You need to modify the script for e.g. the name of your reservation, number of nodes, batch_size, etc. Also, choose either singularity or enroot.
 ```
 cd /proj/nsc/xuan/ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet
 sbash benchmark_nnunet_pytorch_berzelius.sh
-- 
GitLab