Skip to content

P
PGL

PaddlePaddle / PGL

通知 76
Star 4
Fork 0
P
PGL
未验证 提交 958e2aef 编写于 作者: Webbley's avatar Webbley 提交者： GitHub
浏览文件
操作

Merge pull request #1 from PaddlePaddle/master 

fix CUDAPlace
上级 2a7bf171 11604c4e
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
docs/source/quick_start/md/quick_start.md
浏览文件 @ 958e2aef
...@@ -49,7 +49,7 @@ Currently our PGL is developed based on static computational mode of paddle (we ...@@ -49,7 +49,7 @@ Currently our PGL is developed based on static computational mode of paddle (we
import paddle.fluid as fluid import paddle.fluid as fluid
use_cuda = False 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 # use GraphWrapper as a container for graph data to construct a graph neural network
gw = pgl.graph_wrapper.GraphWrapper(name='graph', gw = pgl.graph_wrapper.GraphWrapper(name='graph',
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或