Reusing oversized pinned memory buffers with PyTorch

后端 未结 0 960
眼角桃花
眼角桃花 2020-12-18 04:29

I\'m trying to improve the performance of my PyTorch code by using a preallocated page locked tensor to copy all my results from the GPU into. The results from the GPU will

相关标签:
回答
  • 消灭零回复
提交回复
热议问题