Lorenz data creation script can not use GPUs

There should be a --device keyword so you can run the data-creation on a GPU.