From 6120be45db43f218e2de42cb50cf4f9762ed6d83 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Wed, 11 Oct 2023 16:08:18 +0000
Subject: [PATCH] Update 2 files
- /MLPerf/training/image_segmentation/pytorch/U-Net3D.def
- /MLPerf/training/image_segmentation/pytorch/build.txt
---
.../image_segmentation/pytorch/U-Net3D.def | 25 +++++++++++++++++++
.../image_segmentation/pytorch/build.txt | 15 +++++------
2 files changed, 31 insertions(+), 9 deletions(-)
create mode 100644 MLPerf/training/image_segmentation/pytorch/U-Net3D.def
diff --git a/MLPerf/training/image_segmentation/pytorch/U-Net3D.def b/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
new file mode 100644
index 0000000..8771fe0
--- /dev/null
+++ b/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
@@ -0,0 +1,25 @@
+Bootstrap: localimage
+From: /proj/nsc_testing/xuan/containers/pytorch_1.7.1-cuda11.0-cudnn8-runtime.sif
+
+
+%environment
+
+export PYTHONNOUSERSITE=True
+
+
+%post
+
+mkdir -p /workspace/unet3d
+cd /workspace/unet3d
+
+apt-get update && \
+ apt-get upgrade -y && \
+ apt-get install -y git
+apt-get install -y vim
+
+git clone https://github.com/mlperf/training.git
+mv training/image_segmentation/pytorch/* /workspace/unet3d/
+rm -rf training
+
+pip install --upgrade pip
+pip install --disable-pip-version-check -r requirements.txt
\ No newline at end of file
diff --git a/MLPerf/training/image_segmentation/pytorch/build.txt b/MLPerf/training/image_segmentation/pytorch/build.txt
index 1a679da..fd77634 100644
--- a/MLPerf/training/image_segmentation/pytorch/build.txt
+++ b/MLPerf/training/image_segmentation/pytorch/build.txt
@@ -1,16 +1,13 @@
-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
+apptainer build containers/pytorch_1.7.1-cuda11.0-cudnn8-runtime.sif docker://pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime
-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
+MODULE_NAME=U-Net3D
+# MODULE_VERSION=21.11.0
+CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}.sif
+DEF_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/MLPerf/training/image_segmentation/pytorch/$MODULE_NAME.def
apptainer build $CONTAINER_DIR $DEF_DIR
-# Known issue
-https://github.com/NVIDIA/DeepLearningExamples/issues/1113
+
\ No newline at end of file
--
GitLab