【解决torch.nn.attention找不到问题】ModuleNotFoundError: No module named ‘torch.nn.attention‘

繁华落尽，寻一世真情

于 2025-03-24 23:43:56 发布

阅读量304

点赞数 1

文章标签： pytorch

本文链接：https://blog.csdn.net/weixin_43509698/article/details/146489113

版权

pytorch=2.2.2报torch.nn.attention是由于2.2.2版本还没有nn.attention的功能，需要安装torch<=2.4.1。

本人想偷个懒，在github找到pytorch的源码，将nn目录下的attention目录，拷贝到已有的pytorch安装包目录下。完美解决在pytorch=2.2.2找不到torch.nn.attention模块。

ValueError: f must be a string filename in order to use mmap argument错误，将mmap设置成False

    weights = torch.load(
              ^^^^^^^^^^^
  File "D:\software\anconda\Lib\site-packages\torch\serialization.py", line 1014, in load
    raise ValueError("f must be a string filename in order to use mmap argument")
ValueError: f must be a string filename in order to use mmap argument