Comment 6 for bug 1388217

I suspect the nvidia-uvm module isn't loaded. You can load it manually with 'sudo modprobe nvidia-uvm' before running python. The nvidia-modprobe package loads the necessary modules as needed, see LP: #1361207 .