From 42d7550e353fbd9968c91942ade403679da30a5c Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Wed, 11 Oct 2023 21:59:44 +0000
Subject: [PATCH] Update file benchmark.txt
---
MLPerf/training/image_segmentation/pytorch/benchmark.txt | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/MLPerf/training/image_segmentation/pytorch/benchmark.txt b/MLPerf/training/image_segmentation/pytorch/benchmark.txt
index 6aeed74..d09d8d5 100644
--- a/MLPerf/training/image_segmentation/pytorch/benchmark.txt
+++ b/MLPerf/training/image_segmentation/pytorch/benchmark.txt
@@ -20,7 +20,7 @@ mv kits19/data/* raw-data/
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 --nv -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 "cd /workspace/unet3d && bash run_and_time.sh 1"
+apptainer exec --nv -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"
--
GitLab