Getting a GPU memory error when running the data prep if GPUs are available. Works fine without GPUs.