manimaran_p:

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


@JaimeGGB as @manimaran_p suggested, try following the steps in the section mentioned above (selecting a kernel using a “kernel picker” in VS Code Jupyter notebook)

  
      

      code.visualstudio.com
  

  
    

Working with Jupyter Notebooks in Visual Studio Code 1

  Working with Jupyter Notebooks in Visual Studio Code.


  

  
    
    
  

  






 JaimeGGB:

probably because I don’t know what the environment or the kernel are


Let us know if this helps:

Python environments:


  
      

      code.visualstudio.com
  

  
    

Using Python Environments in Visual Studio Code

  Configuring Python Environments in Visual Studio Code


  

  
    
    
  

  



Kernels:

https://ipython.readthedocs.io/en/stable/install/kernel_install.html

Kernels and envs:


  
      

      Jupyter Community Forum – 8 Aug 21
  

  
    

What is the relationship between kernels and virtual environments?

  I’m confused (noob) about the relationship between IPython kernels and Python virtual environments (at least as they are used in JupyterLab).  For example, if I activate a virtual environment (in my case typically with workon my_venv) and then ...

  
    Reading time: 1 mins 🕑
      Likes: 2 ❤