vscode intellisense not working with PyTorch

Question

I'm using a virtual environment and I have selected the right interpreter.

Now even after I cmd + p > reload window and

import torch.nn as nn

I don't get any suggestion righting nn. on vscode 1.31.1

Answer 1

I found jedi to be quite slow but the recently released Pylance extension works like a charm: https://github.com/microsoft/pylance-release

and then update your settings

"python.languageServer": "Pylance"

Answer 2

Faced the same problem with vscode 1.42. Resolved the problem by going to Settings, Changing Python: Language server from Microsoft to Jedi. Now I can see autocomplete with torch.nn.

Answer 3

Using the beta version of the Python language server (0.2.17), I get completions.

To turn on the language server, follow these instructions. If the stable version doesn't work then follow those instructions on how to download the beta version of the extension to get the completions.