我仅使用 pip install torch 就安装了 PyTorch。我还安装了正确版本的 python(我没有两个不同的版本)。当我在 VS code 中运行以下命令时,它返回了正确的版本,当我检查 PyTorch 是否已通过 pip 安装时,它可以正常工作。
import torch
print(torch.__version__)
Run Code Online (Sandbox Code Playgroud)
但由于某种原因,当我尝试导入 torch,或者尝试从类中的 nn.Module 继承时,VS code 无法识别 torch。我刚刚收到错误“无法解析导入火炬”和“nn 未定义”
我真的很困惑该怎么做,因为我找不到其他人遇到这个问题,而且我看到的所有 PyTorch VS 代码示例都只是安装了 python 扩展,没有问题。