add gpu support

8 years ago · 3648165b63
parent 0e300f9bf0
commit 3648165b63
1 changed files with 2 additions and 0 deletions
--- 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()