From eed248f8a1140c3fb8fa64c8d28af4c55a8bf2b8 Mon Sep 17 00:00:00 2001
From: Christian Kniep <53055915+kniec@users.noreply.github.com>
Date: Fri, 3 Jul 2020 10:26:08 +0200
Subject: [PATCH] Add generic ENV to Dockerfile

---
 aws-batch-using-nice-dcv.dockerfile | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/aws-batch-using-nice-dcv.dockerfile b/aws-batch-using-nice-dcv.dockerfile
index b0eadcc..9ba3bae 100644
--- a/aws-batch-using-nice-dcv.dockerfile
+++ b/aws-batch-using-nice-dcv.dockerfile
@@ -73,6 +73,11 @@ RUN echo "/usr/local/bin/startup_script.sh" >> "/etc/rc.local" \
              "/usr/local/bin/send_dcvsessionready_notification.sh" \
              "/usr/local/bin/startup_script.sh"
 
+ENV DISPLAY=":0"
+ENV NVIDIA_VISIBLE_DEVICES="0"
+ENV NVIDIA_ALL_CAPABILITIES="all"
+
+
 EXPOSE 8443
 
 CMD ["/usr/local/bin/run_script.sh"]