Attempt to train on the GPU.