title: PyTorch: Render point cloud without CPU transfer?
body: <p>I have a some data in pytorch on the GPU, say,</p>
  
  <pre><code>x = torch.cuda.FloatTensor(500, 3).uniform_()
  </code></pre>
  
  <p>Is there any way to render this as a point cloud (ala <a href="https://matplotlib.org/gallery/mplot3d/scatter3d.html" rel="nofollow noreferrer">https://matplotlib.org/gallery/mplot3d/scatter3d.html</a>) without first transferring the data to the CPU?</p>
  
  <p>I have looked at frameworks such as matplotlib, VisPy, mayavi, etc. But all of them seem to want data via the CPU and cannot handle a GPU pointer, say.</p>
  
  <p>Does anyone know of a good method?
  Preferably, the plotting should be non-blocking such that I can have a live view of <code>x</code> during training. Any ideas?</p>