安装了 PyTorch 但 VS code 不会导入 torch

Question

我仅使用 pip install torch 就安装了 PyTorch。我还安装了正确版本的 python（我没有两个不同的版本）。当我在 VS code 中运行以下命令时，它返回了正确的版本，当我检查 PyTorch 是否已通过 pip 安装时，它可以正常工作。

import torch
print(torch.__version__)

但由于某种原因，当我尝试导入 torch，或者尝试从类中的 nn.Module 继承时，VS code 无法识别 torch。我刚刚收到错误“无法解析导入火炬”和“nn 未定义”

我真的很困惑该怎么做，因为我找不到其他人遇到这个问题，而且我看到的所有 PyTorch VS 代码示例都只是安装了 python 扩展，没有问题。

Answer 1

检查 vscode 是否使用与安装 pytorch 相同的 python 解释器和环境。点击 cmd + shift + P 并搜索解释器。单击“Python 解释器”并选择正确的解释器。检查下图所示的图像