Solved by miladiouss in post #18 
              
              
                The solution can be found here. Basically, run the following commands in the terminal: 
sudo rmmod nvidia_uvm
sudo rmmod nvidia
sudo modprobe nvidia
sudo modprobe nvidia_uvm