flash-attn安装

conda activate _________
python - <<'PY'
import sys, torch, importlib.metadata as md
print("python     :", sys.version.split()[0])
print("torch      :", torch.__version__)
print("torch cuda :", torch.version.cuda)
print("torch abi  :", "TRUE" if torch._C._GLIBCXX_USE_CXX11_ABI else "FALSE")
try:
    print("flash-attn :", md.version("flash-attn"))
except Exception as e:
    print("flash-attn : NOT INSTALLED ->", e)
PY