feat: add --user parameter to container run.sh script

Signed-off-by: Keiven Chang <keivenchang@users.noreply.github.com>

Author: keivenchang

container/run.sh

@@ -43,6 +43,7 @@ INTERACTIVE=
 USE_NIXL_GDS=
 RUNTIME=nvidia
 WORKDIR=/workspace
+USER=
 
 get_options() {
     while :; do
@@ -124,6 +125,14 @@ get_options() {
                 missing_requirement "$1"
             fi
             ;;
+        --user)
+            if [ "$2" ]; then
+                USER="$2"
+                shift
+            else
+                missing_requirement "$1"
+            fi
+            ;;
         --privileged)
             if [ "$2" ]; then
                 PRIVILEGED=$2
@@ -274,6 +283,12 @@ get_options() {
         RM_STRING=" --rm "
     fi
 
+    if [[ ${USER} == "" ]]; then
+        USER_STRING=""
+    else
+        USER_STRING="--user ${USER}"
+    fi
+
     if [ -n "$USE_NIXL_GDS" ]; then
         VOLUME_MOUNTS+=" -v /run/udev:/run/udev:ro "
         NIXL_GDS_CAPS="--cap-add=IPC_LOCK"
@@ -310,6 +325,7 @@ show_help() {
     echo "  [-- stop processing and pass remaining args as command to docker run]"
     echo "  [--workdir set the working directory inside the container]"
     echo "  [--runtime add runtime variables]"
+    echo "  [--user override the user for running the container]"
     exit 0
 }
 
@@ -347,6 +363,7 @@ ${RUN_PREFIX} docker run \
     ${NIXL_GDS_CAPS} \
     --ipc host \
     ${PRIVILEGED_STRING} \
+    ${USER_STRING} \
     ${NAME_STRING} \
     ${ENTRYPOINT_STRING} \
     ${IMAGE} \