安装 PyTorch 但 VS 代码不会导入火炬

Question

我已经使用 pip install torch 安装了 PyTorch。我还安装了正确版本的 python（我没有两个不同的版本）。 当我 运行 VS 代码中的以下内容时，它返回了正确的版本，当我检查 PyTorch 是否使用 pip 安装时，它可以正常工作。

import torch
print(torch.__version__)

但出于某种原因，当我尝试导入 torch 或尝试从 nn.Module 继承 class 时，VS 代码无法识别它。 我只是收到错误“无法解析导入火炬”和“未定义 nn”

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

Answer 1

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