找不到名为torch.nn的模块

Question

我正在关注聊天机器人上的一个教程：https://pytorch.org/tutorials/beginner/chatbot_tutorial.html

我有Windows 10。

导入火炬 #works

  

但以下导入无效：

导入torch.nn

导入torch.jit

  

错误：

no module named torch.nn found

and


No module named torch.jit found

  

我已经在anaconda中安装了以下内容：

conda install pytorch-cpu torchvision-cpu -c pytorch

我应该怎么做才能消除错误

代码是：

import torch
from torch.jit import script, trace
import torch.nn as nn
from torch import optim
import torch.nn.functional as F

Answer 1

在那里。 您必须卸下割炬。对于nn，该命令将变为：

import nn as n

import nn.functional as F

卸下割炬。从命令中

Answer 2

万一到这里的其他人犯了和我一样的错误，我为我的 PyTorch 相关代码创建了一个文件夹，我巧妙地将其命名为 torch。这导致了问题，因为未导入 torch python 包，而是导入了本地目录。