A total noob here, but my hunch is that the code takes too long to run on the GPU, and is thus terminated for that reason. I did some searching on google to verify my hunch, and the results seemed encouraging enough for me to post this answer.

Look at 2etime’s YouTube channel. There should be enough there to get you going and then some. edit: this was meant to be an answer to another post, please delete or disregard this. Apologies. It seems that I can’t delete my own posts.