1、基础环境准备
已安装anaconda并创建相应的conda虚拟环境
安装pytorch版本对应的CUDA和cuDNN,注意版本对应关系
2、下载指定版本的xformers安装包
从上述链接下载本地环境对应python版本、cuda版本的whl文件
3、安装xformers
在虚拟conda环境下使用命令安装,例如:”
pip install .\xformers-0.0.22.post7+cu118-cp38-cp38-win_amd64.whl --no-deps“注意:上述命令”--no-deps“ 是为了只安装xformers,否则会改变本地的pytorch环境