安装了 PyTorch 但 VS 代码不会导入 Torch答案

【问题标题】：Installed PyTorch but VS code wont import torch安装了 PyTorch 但 VS 代码不会导入 Torch
【发布时间】：2021-11-18 18:35:36
【问题描述】：

我只使用 pip install torch 安装了 PyTorch。我还安装了正确版本的 python（我没有两个不同的版本）。当我在 VS 代码中运行以下命令时，它返回了正确的版本，并且当我检查 PyTorch 是否与 pip 一起安装时，它可以工作。

import torch
print(torch.__version__)

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

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

【问题讨论】：

您的 Python 在 VS Code 中设置的环境是否与您在运行 pip 的终端中设置的环境相同？ VS Code 在状态栏左侧显示其当前的 Python 环境。点击那里将允许您更改它。

【解决方案1】：

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

【讨论】：