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”