From 765963ab151b71aef8124b26f2bce7fad2e999ed Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuagu37@gmail.com>
Date: Thu, 6 Oct 2022 15:11:28 +0200
Subject: [PATCH] Update README.md

---
 README.md | 29 +++++++++++++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 450bf88..565a147 100644
--- a/README.md
+++ b/README.md
@@ -1,9 +1,34 @@
 # Benchmark_nnU-Net_for_PyTorch
 Benchmark of nnU-Net for PyTorch on Berzelius
 
-## Clone the repository.
+### On local
+- 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 .
+```
+
+- Push the container to Docker Hub
+```
+docker tag nnunet:latest xuagu37/nvidia_nnu-net_for_pytorch:21.11.0
+docker push xuagu37/nvidia_nnu-net_for_pytorch:21.11.0
+```
+
+### On Berzelius
+
+- Clone the repository.  
+```
+git clone https://github.com/NVIDIA/DeepLearningExamples
+cd ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet
+```
+
+- On Berzelius, pull the image for Singularity and run
+```
+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 
+```
-- 
GitLab