#https://medium.com/@Spritan/dealing-with-cuda-initialization-error-aa7c88d021e4
sudo rmmod nvidia_uvm
sudo modprobe nvidia_uvm