【踩坑记录】from natten import NeighborhoodAttention1D

在配置forecast-mae环境的时候。

./forecast-mae/src/model/layers/agent_embedding.py文件中

导入包的时候from natten import NeighborhoodAttention1D

报错，

解决办法，默认安装的是__version__ = "0.14.2"版本的

出现这个问题可能是包损坏所致，

解决办法：可以重新安装 ，可以由低到高版本尝试，重装natten。

 pip install natten==0.14.2
 pip install natten==0.14.4
 pip install natten==0.14.6

关于natten的介绍：

NATTEN is an extension to PyTorch, which provides the first fast sliding window attention with efficient CUDA kernels. It provides Neighborhood Attention (local attention) and Dilated Neighborhood Attention (sparse global attention, a.k.a. dilated local attention) as PyTorch modules for both 1D and 2D data.

NATTEN是PyTorch的扩展，它提供了第一个快速滑动窗口关注与高效的CUDA内核。它作为PyTorch模块，为1D和2D数据提供邻域注意(局部注意)和扩展邻域注意(稀疏全局注意，又称扩展局部注意)。

不少人在配置forecast-mae环境的时候遇到这个问题，这里我将我的环境备份出来给给大家。可以去本文绑定的资源处下载。

forecast-mae运行环境yaml资源-CSDN文库

forecast-mae源代码（包含我的运行环境yaml文件）Yolumia/forecast-mae (github.com)