// 编译https://github.com/thu-ml/SpargeAttn/需要如下东西 apt install nvidia-cuda-toolkit apt install build-essential apt install ninja-build apt-get install cmake apt install python3-dev // 编译时设置环境变量 export TORCH_CUDA_ARCH_LIST="8.0"
