Solved by jmandivarapu1 in post #9 
              
              
                Fixed it. The Main reason is the once you update your NVIDIA GPU drivers it needs us to reinstall the CUDA again. 
we need to uninstall the cuda, pytorch and install cuda, pytorch  again then it works.