[linux] AttributeError: module ‘transformer_engine‘ has no attribute ‘pytorch‘

[BUG] AttributeError: module 'transformer_engine' has no attribute 'pytorch' · Issue #696 · NVIDIA/Megatron-LM · GitHub

其中这个答案并没有解决我的问题:

import flash_attn_2_cuda as flash_attn_cuda

Traceback (most recent call last):

File "<stdin>", line 1, in <module>

ImportError: /mnt/nas/kexin/cv/Open-Sora/xformers/third_party/flash-attention/flash_attn_2_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN3c105ErrorC2ENS_14SourceLocationENSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE

import的flash_attn是xformers的包。。。

cd到目录下 /mnt/nas/kexin/cv/Open-Sora/xformers/third_party/flash-attention/flash_attn_2_cuda.cpython-310-x86_64-linux-gnu.so。

重新安装xformers

报错:

Could not build wheels for xformers, which is required to install pyproject.toml-based projects

复制代码
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu121
相关推荐
看到我请叫我铁锤8 小时前
vue3中THINGJS初始化步骤
前端·javascript·vue.js·3d
q***25218 小时前
SpringMVC 请求参数接收
前端·javascript·算法
q***33378 小时前
Spring Boot项目接收前端参数的11种方式
前端·spring boot·后端
MarkHD8 小时前
车辆TBOX科普 第45次
java·开发语言
烛阴9 小时前
从`new()`到`.DoSomething()`:一篇讲透C#方法与构造函数的终极指南
前端·c#
还债大湿兄9 小时前
阿里通义千问调用图像大模型生成轮动漫风格 python调用
开发语言·前端·python
谢尔登9 小时前
defineProperty如何弥补数组响应式不足的缺陷
前端·javascript·vue.js
鸭子程序员9 小时前
c++ 算法
开发语言·c++·算法
搬砖ing换来金砖9 小时前
Python入门-Task02
开发语言·python
蓝瑟忧伤9 小时前
前端技术新十年:从工程体系到智能化开发的全景演进
前端