在模型库中的Demo,我想使用CPU训练,需要修改哪些代码呢?- PaddlePaddle 飞桨 FAQ合集 - 模型类问题2

PaddlePaddle 飞桨 FAQ合集 - 模型类问题2

Question:

在模型库中的Demo,我想使用CPU训练,需要修改哪些代码呢?

Answer:

将训练/预测文件中的"CUDAPlace(x)"代码改成"CPUPlace()"即可执行CPU运算。因为CPU运算效率太低,推荐使用GPU运算,如果手中没有GPU则可以尝试使用AI Studio进行GPU运算。

你可能感兴趣的:(在模型库中的Demo,我想使用CPU训练,需要修改哪些代码呢?- PaddlePaddle 飞桨 FAQ合集 - 模型类问题2)