Python PyTorch：渲染点云而不传输CPU？

我在GPU上的pytorch中有一些数据

x = torch.cuda.FloatTensor(500, 3).uniform_()

是否有任何方法可以在不首先将数据传输到CPU的情况下将其渲染为点云（ala）

我看过matplotlib、VisPy、mayavi等框架，但它们似乎都希望通过CPU获得数据，无法处理GPU指针

有人知道一个好方法吗？

---

最好，绘图应该是非阻塞的，这样我可以在培训期间实时查看

x

。有什么想法吗？

我无法回答您的问题，但我很好奇：向cpu的传输是否是一个瓶颈？我可以想象，绘图本身需要花费大量的时间，相比之下，将张量传输到cpu是可以忽略的。正如你所说，cpu上的瓶颈肯定是绘图。但是，由于数据已经在GPU上，人们应该能够在屏幕上快速显示它使用例如OpenGL。。。我只是希望有一些Python库可以用来做这件事。