Tensorflow 如何在gpu中直接转换不同深度学习框架的数据(张量),而无需复制到cpu?

Tensorflow 如何在gpu中直接转换不同深度学习框架的数据(张量),而无需复制到cpu?,tensorflow,pytorch,tensor,mxnet,Tensorflow,Pytorch,Tensor,Mxnet,例如,我在pytorch中有一个cuda tensor,如何在gpu中将其直接转换为mxnet/tensorflow,而不将其复制到cpu?经常在cpu和gpu之间复制数据非常耗时。最快的方法就是使用它 只有CPU支持通过PyTorch共享张量 如果你想做一些Python展示,你会因为GIL而慢下来。我现在没有@古旺。

例如,我在pytorch中有一个cuda tensor,如何在gpu中将其直接转换为mxnet/tensorflow,而不将其复制到cpu?经常在cpu和gpu之间复制数据非常耗时。

最快的方法就是使用它

只有CPU支持通过PyTorch共享张量


如果你想做一些Python展示,你会因为GIL而慢下来。

我现在没有@古旺。