From a158bd9173d745b4969e39fcc1c00681a791d251 Mon Sep 17 00:00:00 2001
From: Yancey1989 <yancey1989@gmail.com>
Date: Tue, 5 Jun 2018 17:15:18 +0800
Subject: [PATCH] fix ci

---
 python/paddle/fluid/tests/unittests/test_dist_mnist.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/paddle/fluid/tests/unittests/test_dist_mnist.py b/python/paddle/fluid/tests/unittests/test_dist_mnist.py
index e6bc56cce..e4d39c8b3 100644
--- a/python/paddle/fluid/tests/unittests/test_dist_mnist.py
+++ b/python/paddle/fluid/tests/unittests/test_dist_mnist.py
@@ -27,6 +27,7 @@ import signal
 
 SEED = 1
 DTYPE = "float32"
+paddle.dataset.mnist.fetch()
 
 
 # random seed must set before configuring the network.
@@ -147,7 +148,7 @@ class TestDistMnist(unittest.TestCase):
         os.kill(pid, signal.SIGTERM)
 
     def test_with_place(self):
-        p = fluid.CUDAPlace() if core.is_compiled_with_cuda(
+        p = fluid.CUDAPlace(0) if core.is_compiled_with_cuda(
         ) else fluid.CPUPlace()
 
         pserver_pid = self.start_pserver(self._ps_endpoints)
-- 
GitLab