From 3c6361e8ddc77e55003dc3c4fb9ad60032b8e3fb Mon Sep 17 00:00:00 2001
From: Hai Liang Wang <hailiang.hl.wang@gmail.com>
Date: Tue, 4 Apr 2017 05:46:17 -0700
Subject: [PATCH] #4 run with nvidia-docker

---
 scripts/train_with_docker.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/train_with_docker.sh b/scripts/train_with_docker.sh
index 71ac445..da70a73 100755
--- a/scripts/train_with_docker.sh
+++ b/scripts/train_with_docker.sh
@@ -16,7 +16,7 @@ function printUsage(){
 
 function main() {
     cd $rootDir
-    docker run --name deepqa2  \
+    nvidia-docker run --name deepqa2  \
         -v $rootDir/save:/deepqa2/save \
         -v $rootDir/data:/deepqa2/data \
         -v $rootDir/config.ini:/deepqa2/config.ini \