Is there any update on this?
I’m working with deep reinforcement learning on the petting zoo 1 environment, specifically the MPE particle system. I notice how the rendering function in the system uses pyglet and OpenGL. It uses the latter to create the buffer image and then copies it to a numpy array which is pretty slow. I was wondering if you could grab the array directly from the GPU with a torch tensor. Is it possible without accessing C++ code?