This is in no way a definitive guide to run PyTorch with Nvidia container along with VS Code. But it works. For now. Code from PyTorch tutorial.
X forwarding: Use command xhost +local:docker
in the host to allow X forwarding, if not running the code from VS code. initializeCommand
will take care of this for VS Code.