add gpu support

3648165b · qiaolongfei · 0e300f9b · 3648165b
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

python/paddle/v2/framework/tests/mnist.py python/paddle/v2/framework/tests/mnist.py +2 -0

未找到文件。
--- a/python/paddle/v2/framework/tests/mnist.py
+++ b/python/paddle/v2/framework/tests/mnist.py
@@ -7,6 +7,8 @@ BATCH_SIZE = 100

 scope = core.Scope()
 place = core.CPUPlace()
+# if you want to test GPU training, you can use gpu place
+# place = core.GPUPlace(0)
 dev_ctx = core.DeviceContext.create(place)

 init_net = core.Net.create()