飞桨百度架构师手把手带你零基础实践深度学习——手写数字识别 资源配置

使用GPU训练

使用GPU进行训练

1. CPU：fluid.CPUPlace()
2. GPU: fluid.CUDAPlace(0)，四个GPU卡的编号0,1,2,3

#仅前3行代码有所变化，在使用GPU时，可以将use_gpu变量设置成True
use_gpu = False
place = fluid.CUDAPlace(0) if use_gpu else fluid.CPUPlace()

with fluid.dygraph.guard(place):  #用place作为资源进行训练
    model = MNIST()
    model.