2D classification only run on GPU 0 or GPU 1 not on both GPUs