首页 Paddle框架 帖子详情
Paddle框架GPU下共享Tensor的问题
收藏
快速回复
Paddle框架 问答模型训练 253 0
Paddle框架GPU下共享Tensor的问题
收藏
快速回复
Paddle框架 问答模型训练 253 0

请问paddle在GPU环境下多线程编程,如何实现Tensor的共享

在pytorch框架中可以通过torch.multiprocessing模块来实现,只需要在需要的tensor里引用share_memory()方法就好了。

但是我没有看到Paddle中有类似的方法能实现。

请问有别的方法来替代吗?

0
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户