From 5a8ae7efdd78af2e0094ea240c70339f335dc87a Mon Sep 17 00:00:00 2001 From: Webbley Date: Mon, 13 Apr 2020 20:14:30 +0800 Subject: [PATCH] fix bug of CUDAPlace in docs --- docs/source/quick_start/md/quick_start.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/quick_start/md/quick_start.md b/docs/source/quick_start/md/quick_start.md index de50b83..7df4f48 100644 --- a/docs/source/quick_start/md/quick_start.md +++ b/docs/source/quick_start/md/quick_start.md @@ -49,7 +49,7 @@ Currently our PGL is developed based on static computational mode of paddle (we import paddle.fluid as fluid use_cuda = False -place = fluid.GPUPlace(0) if use_cuda else fluid.CPUPlace() +place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace() # use GraphWrapper as a container for graph data to construct a graph neural network gw = pgl.graph_wrapper.GraphWrapper(name='graph', -- GitLab