From a0f2925718aaafdfb6e176b6ef12885f2be17202 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Wed, 11 Oct 2023 21:22:03 +0000
Subject: [PATCH] Update 3 files

- /MLPerf/training/image_segmentation/pytorch/benchmark.txt
- /MLPerf/training/image_segmentation/pytorch/U-Net3D.def
- /MLPerf/training/image_segmentation/pytorch/build.txt
---
 MLPerf/training/image_segmentation/pytorch/U-Net3D.def   | 2 +-
 MLPerf/training/image_segmentation/pytorch/benchmark.txt | 2 +-
 MLPerf/training/image_segmentation/pytorch/build.txt     | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/MLPerf/training/image_segmentation/pytorch/U-Net3D.def b/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
index c145a55..e60aa38 100644
--- a/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
+++ b/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
@@ -1,5 +1,5 @@
 Bootstrap: localimage
-From: /proj/nsc_testing/xuan/containers/pytorch_1.7.1-cuda11.0-cudnn8-runtime.sif
+From: /proj/nsc_testing/xuan/containers/pytorch_1.7.1-cuda11.0-cudnn8-devel.sif
 
 
 %environment
diff --git a/MLPerf/training/image_segmentation/pytorch/benchmark.txt b/MLPerf/training/image_segmentation/pytorch/benchmark.txt
index 65953d5..6aeed74 100644
--- a/MLPerf/training/image_segmentation/pytorch/benchmark.txt
+++ b/MLPerf/training/image_segmentation/pytorch/benchmark.txt
@@ -22,5 +22,5 @@ rm -rf kits19
 
 apptainer exec -B ${WORK_DIR}/raw-data:/raw-data -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results $CONTAINER_DIR bash -c "cd /workspace/unet3d && python3 preprocess_dataset.py --data_dir /raw-data --results_dir /data"
 
-apptainer exec -B ${WORK_DIR}/raw-data:/raw-data -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results $CONTAINER_DIR bash -c "bash /workspace/unet3d/run_and_time.sh 1"
+apptainer exec -B ${WORK_DIR}/raw-data:/raw-data -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results $CONTAINER_DIR bash -c "cd /workspace/unet3d && bash run_and_time.sh 1"
 
diff --git a/MLPerf/training/image_segmentation/pytorch/build.txt b/MLPerf/training/image_segmentation/pytorch/build.txt
index fd77634..59ad0a5 100644
--- a/MLPerf/training/image_segmentation/pytorch/build.txt
+++ b/MLPerf/training/image_segmentation/pytorch/build.txt
@@ -1,5 +1,5 @@
 
-apptainer build containers/pytorch_1.7.1-cuda11.0-cudnn8-runtime.sif  docker://pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime
+apptainer build containers/pytorch_1.7.1-cuda11.0-cudnn8-devel.sif  docker://pytorch/pytorch:1.7.1-cuda11.0-cudnn8-devel
 
 
 MODULE_NAME=U-Net3D
-- 
GitLab