请务必注意,为了能用使用xformers加速训练,python版本务必使用3.10。此外,xformers版本与torch版本务必匹配:
https://blog.csdn.net/BigerBang/article/details/139685883
https://blog.csdn.net/BigerBang/article/details/140301608
另外,还需要安装triton,现有的源中没有这个库,因此需要手动下载并安装:
https://pythonziliao.com/post/3106.html