远程角色会生成经验(存储在GPU上的cuda张量),是否可以在不复制回CPU的情况下对其进行采样和使用?

时间:2019-02-01 00:39:36

标签: pytorch ray

我使用Ray远程演员生成重播体验,然后对一些体验进行采样,并将其返回到主要功能,然后将其传递给RL代理进行培训。我可以在GPU上使用它们,并且避免复制回CPU然后再复制到GPU吗?

0 个答案:

没有答案