For the world model training I've used an RTX 4060, but for the agent playing the game that was directly on the pico using stream-ac