trying to run cifar-10 on GPU, not there yet