1:查看默认运行的device设备
import paddle
print(paddle.device.get_device()) # 默认是GPU0
2:指定GPU,通过set_device函数接口
import paddle
print(paddle.device.set_device('gpu:1'))
print(paddle.device.get_device())
示例:
import paddle
# 指定在 CPU 上训练
paddle.device.set_device('cpu')
# 指定在 GPU 第 0 号卡上训练
# paddle.device.set_device('gpu:0')
参考链接
1: 模型训练、评估与推理-使用文档-PaddlePaddle深度学习平台
2: 分布式训练-使用文档-PaddlePaddle深度学习平台
本文转载自: https://blog.csdn.net/lilai619/article/details/128579048
版权归原作者 微风❤水墨 所有, 如有侵权,请联系我们删除。
版权归原作者 微风❤水墨 所有, 如有侵权,请联系我们删除。