Initial setup for GPU runs