tensorpipe icon indicating copy to clipboard operation
tensorpipe copied to clipboard
verified publisher icon pytorch

Support GPU tensor transport on pytorch

Open Jason-Khan opened this issue 5 years ago • 1 comments

According to https://pytorch.org/docs/stable/rpc.html, tensorpipe currently does not support sending GPU tensors over the wire. I'm wondering when this functionality would be implemented and integrated with pytorch.

Jason-Khan avatar Aug 11 '20 06:08 Jason-Khan

GPU support is what we're currently working on. You can see that #175 adds a first native CUDA channel. More changes will be needed after that, but we're hoping to get this all done in time for PyTorch 1.7.

lw avatar Aug 11 '20 07:08 lw