安装要求

硬件

• 支持 CUDA 的 NVIDIA GPU（用于 GPU 特定测试和基准测试）。

软件

• Python 3.10 或更高版本

• PyTorch（支持 CUDA）

• Triton

• CuPy（cupy-cuda12x）