Well, the GPU seems to be working again after fighting with my computer for a full day or so. Before running any scripts using the GPU, be sure to

'sudo modprobe nvidia_340_uvm'

and everything should work. Otherwise, I could just turn the GPU off.

Got another problem now. What in the everloving fuck am I doing when calculating
the nKr array? Why the hell do I somehow think I need to calculate it for all n?
FIX THAT SHITE.