Paddle 1.8版本中如何启用gpu?

wixjitnu  于 2021-12-07  发布在  Java
关注(0)|答案(4)|浏览(758)

1.8版本中如何启用gpu?这个项目“建议使用GPU来运行”,请问是在何处设置了启用gpu?
https://aistudio.baidu.com/aistudio/projectdetail/1455734?forkThirdPart=1

0yycz8jy

0yycz8jy1#

"No one assigned"是什么意思?

j2qf4p5b

j2qf4p5b2#

我们无法访问你的 aistudio 代码, 因为它是私密的.

开启 austudio 示例的时候,可以选择 cpu 还是 gpu, 然后代码中是可以通过 place 的方式使用 gpu.

动态图

place = paddle.fluid.CUDAPlace(0)
paddle.fliud.dygraph.enable_dygraph(place)

静态图

place = paddle.fluid.CUDAPlace(0)
exe = paddle.fluid.Executor(place)

exe.run(.....)
pkmbmrz7

pkmbmrz74#

这个需要看 train.py 里面的代码.

def train():
    """
训练主体
:return:
"""
    # 会自动根据当前 paddle 是CPU版本还是GPU版本选择运行硬件
    # 如果是 GPU,默认使用第 0 块
    # 如果希望指定使用,需要主动传入 place 变量,或者通过设置 CUDA_VISIBLE_DEVICES 环境变量控制可见显卡
    utils.logger.info("start train")
    with fluid.dygraph.guard():

fluid.dygraph.guard() 这里,如果没有传 place 则有 gpu 自动用第一张显卡, 否则用 cpu.

相关问题