Skip to content
Snippets Groups Projects
Unverified Commit 765963ab authored by Xuan Gu's avatar Xuan Gu Committed by GitHub
Browse files

Update README.md

parent 06d890aa
No related branches found
No related tags found
No related merge requests found
# Benchmark_nnU-Net_for_PyTorch # Benchmark_nnU-Net_for_PyTorch
Benchmark of nnU-Net for PyTorch on Berzelius Benchmark of nnU-Net for PyTorch on Berzelius
## Clone the repository. ### On local
- Clone the repository.
```
git clone https://github.com/NVIDIA/DeepLearningExamples git clone https://github.com/NVIDIA/DeepLearningExamples
cd ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet cd ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet
```
## Build the nnU-Net PyTorch NGC container. - Build the nnU-Net PyTorch NGC container.
```
docker build -t nnunet . 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
```
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment