我目前正在尝试实现一个使用 sentence_transformers 包的 NLP 文本分类任务。导入所述包时出现以下错误:
ModuleNotFoundError:没有名为“torch.ao.nn.intrinsic.modules.fused”的模块
在我看来,火炬扩展存在一些依赖性或下载问题,因为尝试在独立脚本中导入火炬时会引发相同的错误。有没有人遇到过同样的问题或至少有类似的问题?我在这里迷路了。
我试过下载不同版本的火炬包及其扩展,并找到了一些关于需求的信息但没有成功。
提前致谢
---编辑---
我在虚拟环境中使用 python 版本 3.8.10。导入 torch 时的调用栈为:
ModuleNotFoundError Traceback (most recent call last)
Cell In[1], line 1
----> 1 import torch
File c:\path\MTW_Projects\Active_BERTOPIC\venvBERTOPIC\lib\site-packages\torch\__init__.py:1263
1261 # nn.quant* depends on ao -- so should be after those.
1262 import torch.nn.quantizable
-> 1263 import torch.nn.quantized
1264 import torch.nn.qat
1265 import torch.nn.intrinsic
File c:\path\MTW_Projects\Active_BERTOPIC\venvBERTOPIC\lib\site-packages\torch\nn\quantized\__init__.py:1
----> 1 from . import dynamic # noqa: F403
2 from . import functional # noqa: F403
3 from . import modules # noqa: F403
File c:\path\MTW_Projects\Active_BERTOPIC\venvBERTOPIC\lib\site-packages\torch\nn\quantized\dynamic\__init__.py:1
----> 1 from torch.ao.nn.quantized.dynamic import * # noqa: F403
File c:\path\MTW_Projects\Active_BERTOPIC\venvBERTOPIC\lib\site-packages\torch\ao\nn\quantized\__init__.py:1
----> 1 from . import functional
2 from .modules import * # noqa: F403
3 from .modules import MaxPool2d
...
----> 1 from .fused import _FusedModule # noqa: F401
2 from .fused import ConvBn1d
3 from .fused import ConvBn2d
ModuleNotFoundError: No module named 'torch.ao.nn.intrinsic.modules.fused'
确保你在一个环境中工作。 如果您可以显示您的工作环境的快照和有帮助的错误。