https://stackoverflow.com/questions/10871412/resetting-gpu-and-driver-after-cuda-error
sudo rmmod nvidia_uvm sudo modprobe nvidia_uvm