CUDA initialization: CUDA unknown error – this may be due to an incorrectly set up environment

Had the same issue and in my case solution was very easy, however it wasn’t easy to find it. I had to remove and insert nvidia_uvm module. So:

> sudo rmmod nvidia_uvm
> sudo modprobe nvidia_uvm

That’s all. Just before these command collect_env.py reported “Is CUDA available: False”. After: “Is CUDA available: True”