ComfyUI 安装pytorch与xformers版本冲突问题

ComfyUI 大多用GPU来运行，（为啥，因为快啊）， 需要这二个包支持
先装xformers， 再装pytorch， 不然包不兼容，
不兼容
不兼容

起因

一开始我从这里但版本https://pytorch.org/get-started/locally/

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

自动装了最新版的2.51+cu121版

再装xformers时自动把torch改到2.4.0版，torchaudio没有一起改导致版本不一样，用起来会有问题。

解决

安装包

先装最新版的xformers, 自动装上torch cpu版， 需要改为gpu版，且torch 与torchaudio版本要统一

pip install xformers
pip uninstall torch

pip install torch==2.4.0 --index-url https://download.pytorch.org/whl/cu121
pip install torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu121

后面的cu121指CUDA121，可用nvcc --version查看， 具体可以看这里:https://blog.csdn.net/qq_28171389/article/details/143712138

提示无法找到入口，无法定位程序输入点 torchvision_C.pyd

torchvision 这个版本有问题，https://pypi.org/project/torchvision/， 从这里查看torch2.4.0对应的版本号是0.19
删除掉torchvision，再装

pip uninstall torchvision
pip install torchvision==0.19 --index-url https://download.pytorch.org/whl/cu121

最终:

提示：

1. torch， torchaudio, torchvision 后面要加+ cu121（版本根据实际情况来），这样才是GPU
2. torch， torchaudio 版本号要相同，不同用起来可能有问题
3. 可以先装torch, 再装xformers, 然后再装torch ，转来转去都是版本问题

====================================

上面的都可以忽略了， 找到新的方法：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

https://download.pytorch.org/whl/xformers/
下载 xformers-0.0.28.post3-cp311-cp311-win_amd64.whl 根据自已的版本来

pip install xformers-0.0.28.post3-cp311-cp311-win_amd64.whl

这二个包都是最新版的,再也不用担心冲突了

原文地址：https://blog.csdn.net/qq_28171389/article/details/143820731

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！