diff --git a/README.md b/README.md index 450bf88a100c665c22321e9951ea6684791ce757..565a147b4129a5ffda9b140e08a71b57afd4b8f6 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 +```