Skip to content
Snippets Groups Projects
build.txt 891 B
Newer Older
Xuan Gu's avatar
Xuan Gu committed
https://catalog.ngc.nvidia.com/orgs/nvidia/resources/nnunet_for_pytorch/quick-start-guide

Xuan Gu's avatar
Xuan Gu committed
apptainer build containers/nvidia_pytorch_21.11-py3.sif  docker://nvcr.io/nvidia/pytorch:21.11-py3

MODULE_NAME=nnunet_for_pytorch
MODULE_VERSION=21.11.0
Xuan Gu's avatar
Xuan Gu committed
CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}_${MODULE_VERSION}.sif
DEF_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet/${MODULE_NAME}_${MODULE_VERSION}.def
Xuan Gu's avatar
Xuan Gu committed

Xuan Gu's avatar
Xuan Gu committed
apptainer build $CONTAINER_DIR $DEF_DIR
Xuan Gu's avatar
Xuan Gu committed

Xuan Gu's avatar
Xuan Gu committed

Xuan Gu's avatar
Xuan Gu committed

MODULE_NAME=nnunet_for_pytorch
MODULE_VERSION=latest
CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}_${MODULE_VERSION}.sif
Xuan Gu's avatar
Xuan Gu committed
DEF_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/${MODULE_NAME}_${MODULE_VERSION}.def
Xuan Gu's avatar
Xuan Gu committed

apptainer build $CONTAINER_DIR $DEF_DIR


Xuan Gu's avatar
Xuan Gu committed
# Known issue 
https://github.com/NVIDIA/DeepLearningExamples/issues/1113