From f06b64fee47c1d807a224049243d2d3dec39fc5c Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Thu, 22 Dec 2016 07:45:18 +0000
Subject: [PATCH] Test GPU

---
 demo/mnist/api_train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo/mnist/api_train.py b/demo/mnist/api_train.py
index ce75d79bebe..e5a9075c8ec 100644
--- a/demo/mnist/api_train.py
+++ b/demo/mnist/api_train.py
@@ -54,7 +54,7 @@ def input_order_converter(generator):
 
 
 def main():
-    api.initPaddle("-use_gpu=false", "-trainer_count=4")  # use 4 cpu cores
+    api.initPaddle("-use_gpu=true", "-trainer_count=4")  # use 4 cpu cores
     config = paddle.trainer.config_parser.parse_config(
         'simple_mnist_network.py', '')
 
-- 
GitLab