[GH-ISSUE #20] 关于flash_attn #17

Closed
opened 2026-10-26 21:01:42 +00:00 by bsakaguchi · 1 comment
Owner

Originally created by @GXKIM on GitHub (Feb 2, 2024).
Original GitHub issue: https://github.com/deepseek-ai/DeepSeek-MoE/issues/20

torch 2.1
transformers:4.37.1
显卡:A800
手动安装装进去了,但是检测不到flash_attn
直接用pip 装不进去
去github flash_attn 是支持A800的

image
Originally created by @GXKIM on GitHub (Feb 2, 2024). Original GitHub issue: https://github.com/deepseek-ai/DeepSeek-MoE/issues/20 torch 2.1 transformers:4.37.1 显卡:A800 手动安装装进去了,但是检测不到flash_attn 直接用pip 装不进去 去github flash_attn 是支持A800的 <img width="329" alt="image" src="https://github.com/deepseek-ai/DeepSeek-MoE/assets/125621025/bec74bc6-a21a-4ae6-952b-3253051830da">
Author
Owner

@zwd003 commented on GitHub (Feb 4, 2024):

这个是transformers的检测机制问题,可以在源代码目录试试pip install .

<!-- gh-comment-id:1925556601 --> @zwd003 commented on GitHub (Feb 4, 2024): 这个是transformers的检测机制问题,可以在源代码目录试试pip install .
Sign in to join this conversation.
No labels
pull-request
WIP
No milestone
No project
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference
deepseek-ai/DeepSeek-MoE#17
No description provided.