1.8版本中如何启用gpu?这个项目“建议使用GPU来运行”,请问是在何处设置了启用gpu?https://aistudio.baidu.com/aistudio/projectdetail/1455734?forkThirdPart=1
0yycz8jy1#
"No one assigned"是什么意思?
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(.....)
xfb7svmp3#
代码是这个https://aistudio.baidu.com/aistudio/projectdetail/169412?channelType=0&channel=0
麻烦再看看,我的项目和这个相同https://aistudio.baidu.com/aistudio/projectdetail/169412?channelType=0&channel=0
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.
4条答案
按热度按时间0yycz8jy1#
"No one assigned"是什么意思?
j2qf4p5b2#
我们无法访问你的 aistudio 代码, 因为它是私密的.
开启 austudio 示例的时候,可以选择 cpu 还是 gpu, 然后代码中是可以通过 place 的方式使用 gpu.
动态图
静态图
xfb7svmp3#
代码是这个https://aistudio.baidu.com/aistudio/projectdetail/169412?channelType=0&channel=0
麻烦再看看,我的项目和这个相同https://aistudio.baidu.com/aistudio/projectdetail/169412?channelType=0&channel=0
pkmbmrz74#
这个需要看 train.py 里面的代码.
fluid.dygraph.guard() 这里,如果没有传 place 则有 gpu 自动用第一张显卡, 否则用 cpu.