Solved by manimaran_p in post #3 
              
              
                Not sure if you have checked this, try selecting the interpreter(python env with TensorFlow) in the notebook editor, as mentioned here 
Working with Jupyter Notebooks in Visual Studio Code