找不到名为 torch.nn 的模块

Posted 2023-02-16

【中文标题】找不到名为 torch.nn 的模块

【英文标题】：No module named torch.nn found

【发布时间】：2019-08-26 12:58:18

【问题描述】：

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

我有 Windows 10。

import torch #works 

但以下导入不起作用：

import torch.nn
import torch.jit

错误：

no module named torch.nn found

和

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

【问题讨论】：

您最终找到了解决方案或解决方法吗？

不，我结束了你使用 colab

【参考方案1】：

那里。 你必须移除火炬。对于 nn ，命令将变为：

import nn as n

import nn.functional as F

移除火炬。来自命令

【参考方案2】：

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