diff --git a/MLPerf/training/image_segmentation/pytorch/build.txt b/MLPerf/training/image_segmentation/pytorch/build.txt new file mode 100644 index 0000000000000000000000000000000000000000..1a679dabd10f3fb4cb20e1b5054bd66370d207b9 --- /dev/null +++ b/MLPerf/training/image_segmentation/pytorch/build.txt @@ -0,0 +1,16 @@ +https://catalog.ngc.nvidia.com/orgs/nvidia/resources/nnunet_for_pytorch/quick-start-guide + +apptainer build containers/nvidia_pytorch_21.11-py3.sif docker://nvcr.io/nvidia/pytorch:21.11-py3 + +export APPTAINER_BINDPATH= + +MODULE_NAME=nnunet_for_pytorch +MODULE_VERSION=21.11.0 +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 + +apptainer build $CONTAINER_DIR $DEF_DIR + + +# Known issue +https://github.com/NVIDIA/DeepLearningExamples/issues/1113