From 42be2bd2454b09ab45e34a6d82ca07f2704917c0 Mon Sep 17 00:00:00 2001
From: Mabel Zhang <mabel@openrobotics.org>
Date: Thu, 17 Dec 2020 04:54:03 -0500
Subject: [PATCH] enable nvidia runtime and privileged flag for competitor
 container

Signed-off-by: Mabel Zhang <mabel@openrobotics.org>
---
 run_trial.bash | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/run_trial.bash b/run_trial.bash
index 6c4093f..24d6fd4 100755
--- a/run_trial.bash
+++ b/run_trial.bash
@@ -24,6 +24,7 @@ usage()
 }
 
 # Parse arguments
+RUNTIME="runc"
 nvidia_arg=""
 image_nvidia=""
 
@@ -34,6 +35,7 @@ do
 
   case $key in
       -n|--nvidia)
+      RUNTIME="nvidia"
       nvidia_arg="-n"
       image_nvidia="-nvidia"
       shift
@@ -170,6 +172,8 @@ docker run \
     --env ROS_MASTER_URI=${ROS_MASTER_URI} \
     --env ROS_IP=${COMPETITOR_ROS_IP} \
     --ip ${COMPETITOR_ROS_IP} \
+    --privileged \
+    --runtime=$RUNTIME \
     ${DOCKERHUB_IMAGE} &
 
 # Run competition until server is ended