Currently we only use GPU 0 for OptiX, but that need to change. We should pass to init the device on which we want to operate