I installed the repo driver and it's working fine

filename: /lib/modules/4.0.0-kali1-amd64/updates/dkms/nvidia-current.ko
alias: char-major-195-*
version: 352.21
supported: external
license: NVIDIA
alias: pci:v00