Have you tried to use the official VSCode devcontainer:

  
      

      github.com
  

  
    

vscode-dev-containers/repository-containers/github.com/tensorflow/tensorflow/.de... 28

  main/repository-containers/github.com/tensorflow/tensorflow/.devcontainer

  A repository of development container definitions for the VS Code Remote - Containers extension and GitHub Codespaces - vscode-dev-containers/repository-containers/github.com/tensorflow/tensorflow/...