首先需要安装带gpu的paddlepaddle。其次 在训练初始化中设置

with_gpu = os.getenv('WITH_GPU', '1') != '0'
paddle.init(use_gpu=True, trainer_count=1)

为什么我报错

AttributeError: module 'paddle' has no attribute 'init'

可以到这个页面，先安装好对应版本的paddle https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/windows-pip.html