support for specifiction of gpu number in cifar bandit