环境：PyTorch 2.2.1 + cu118

pip install xformers==0.0.25.post1 --index-url https://download.pytorch.org/whl/cu118

这会自动下载一个 为 CUDA 11.8 编译的 GPU 版本（匹配 PyTorch 2.2.1），下次可以试着先下载xFormers，就不用手动下载torch了。

import xformers
import torch
print("xFormers version:", xformers.__version__)
print("CUDA available:", torch.cuda.is_available())

from xformers.ops import memory_efficient_attention
print("xFormers attention op:", memory_efficient_attention)

上方为检查安装成功的代码