added an option to select the GPU device