使用官方 Wheel 安装 NATTEN
安装完 Torch 后,千万不要直接运行 pip install natten。因为 A800 属于 sm_80 或更高架构,从源码编译 natten 会消耗极长的时间(可能超过 30 分钟),且容易因为环境缺失报错。
请直接下载针对 Linux + Python 3.10 + CUDA 12.1 预编译好的包:
pip install natten==0.14.6 --no-build-isolation -i https://pypi.tuna.tsinghua.edu.cn/simple