【Paddle 2.0-alpha】动态图如何指定使用GPU?

wqlqzqxt  于 2022-04-21  发布在  Java
关注(0)|答案(4)|浏览(198)

如题。

bfnvny8b

bfnvny8b2#

可以使用这个接口 https://www.paddlepaddle.org.cn/documentation/docs/zh/2.0-alpha/api_cn/dygraph_cn/guard_cn.html ,指定下 place

我用这样的代码
with fluid.dygraph.guard(fluid.CUDAPlace()):
在AI Studio只要执行就会重启执行器。另外,使用下面的代码也是会重启执行器
paddle.enable_imperative(paddle.fluid.CUDAPlace(0))

hwamh0ep

hwamh0ep3#

"重启执行器" 指的是什么,动态图应该没有执行器的概念了

u91tlkcl

u91tlkcl4#

AI Studio平台里执行,notebook的执行器会重启

相关问题