From a33f4138d0618945f60e770d172d68c974538953 Mon Sep 17 00:00:00 2001
From: Jiangtao Hu <ycool@users.noreply.github.com>
Date: Sat, 2 Sep 2017 11:24:14 -0700
Subject: [PATCH] docker: map nvidia device into docker. (#1460)

---
 docker/scripts/dev_start.sh     | 1 +
 docker/scripts/dev_start_gpu.sh | 3 ++-
 docker/scripts/release_start.sh | 1 +
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/docker/scripts/dev_start.sh b/docker/scripts/dev_start.sh
index 322216eb11..2d0bdb43bb 100755
--- a/docker/scripts/dev_start.sh
+++ b/docker/scripts/dev_start.sh
@@ -99,6 +99,7 @@ function main(){
     devices="${devices} $(find_device can*)"
     devices="${devices} $(find_device ram*)"
     devices="${devices} $(find_device loop*)"
+    devices="${devices} $(find_device nvidia*)"
     USER_ID=$(id -u)
     GRP=$(id -g -n)
     GRP_ID=$(id -g)
diff --git a/docker/scripts/dev_start_gpu.sh b/docker/scripts/dev_start_gpu.sh
index 78bb7b99f0..1366a08f52 100755
--- a/docker/scripts/dev_start_gpu.sh
+++ b/docker/scripts/dev_start_gpu.sh
@@ -73,7 +73,7 @@ function main(){
     #FIX ME: remove login when open source.
     docker login -u autoapollo -p baidu123
     docker pull $IMG
-    
+
     docker ps -a --format "{{.Names}}" | grep 'apollo_dev' 1>/dev/null
     if [ $? == 0 ]; then
         docker stop apollo_dev 1>/dev/null
@@ -97,6 +97,7 @@ function main(){
     devices="${devices} $(find_device can*)"
     devices="${devices} $(find_device ram*)"
     devices="${devices} $(find_device loop*)"
+    devices="${devices} $(find_device nvidia*)"
     USER_ID=$(id -u)
     GRP=$(id -g -n)
     GRP_ID=$(id -g)
diff --git a/docker/scripts/release_start.sh b/docker/scripts/release_start.sh
index 00fe1e85b9..a243262c04 100755
--- a/docker/scripts/release_start.sh
+++ b/docker/scripts/release_start.sh
@@ -94,6 +94,7 @@ function main() {
     devices="${devices} $(find_device can*)"
     devices="${devices} $(find_device ram*)"
     devices="${devices} $(find_device loop*)"
+    devices="${devices} $(find_device nvidia*)"
     local display=""
     if [[ -z ${DISPLAY} ]];then
         display=":0"
-- 
GitLab