Added set_gpu_index method from astra-1.4