From c0f7d44f48613521f02d0850e8731f437ac23ae7 Mon Sep 17 00:00:00 2001
From: "S.P. Mohanty" <spmohanty91@gmail.com>
Date: Thu, 21 Mar 2024 18:14:07 +0000
Subject: [PATCH] add gpu flags

---
 docker_run.sh | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/docker_run.sh b/docker_run.sh
index 7d503da..e79f4d0 100755
--- a/docker_run.sh
+++ b/docker_run.sh
@@ -26,7 +26,14 @@ echo "Total build time: $BUILD_TIME seconds"
 # 'python /submission/local_evaluation.py' is the command executed inside the container.
 # the -w sets the workind directory to /submission.
 # It then local_evaluation.py using software runtime set up in the Dockerfile.
-docker run -v "$(pwd)":/submission -w /submission $IMAGE_NAME python local_evaluation.py
+docker run \
+    --gpus all \
+    -v "$(pwd)":/submission \
+    -w /submission \
+    $IMAGE_NAME python local_evaluation.py
+
+# Note: We assume you have nvidia-container-toolkit installed and configured 
+# to use the --gpus all flag. If you are not using GPUs, you can remove this flag.
 
 
 # Note 1: Please refer to the Dockerfile to understand how the software runtime is set up.
-- 
GitLab