怎么指定cuda版本安装xformers
直接pip安装xformers会默认使用支持该xformers的cuda最高版本,可能和环境的cuda版本不对应。中的方法可以指定cuda版本,我的pytorch时2.3.0问题解决。
·
首先可以在Files | Anaconda.org上查找符合的xformers版本
直接pip安装xformers会默认使用支持该xformers的cuda最高版本,可能和环境的cuda版本不对应
按照安装xFormers时遇到的问题,以及正确的安装方式_xformers安装-CSDN博客中的idex方法可以指定cuda版本:
比如:在cu11.8下安装0.0.26版本的xformers则可以用pip3 install xformers==0.0.26 --index-url https://download.pytorch.org/whl/cu118
问题解决。
更多推荐
所有评论(0)