问题描述
一些新的功能需要更高的cuda驱动版本,但是我们升级完cuda之后,之前的一些配套环境也需要先升级为相匹配的版本。
升级方法
PyTorch
1.安装相匹配的Pytorch轮子:download.pytorch.org/whl/torch/
我们提供的是torch-2.7.0+cu128-cp312-cp312-win_amd64.whl
如果之前有安装需要先通过以下代码卸载:
python.exe -m pip uninstall torch -y
python.exe -m pip uninstall torchvision torchaudio -y
然后按安装下载的轮子(轮子路径为你下载的路径,下面所有代码同)
python.exe -m pip install H:\Software\Xformers\torch-2.7.0+cu128-cp312-cp312-win_amd64.whl
2.下载相匹配的torchvision:https://download.pytorch.org/whl/cu128/torchvision/
我们提供的是torchvision-0.22.1+cu128-cp312-cp312-win_amd64.whl
安装代码
python.exe -m pip install --no-deps H:\Software\Xformers\torchvision-0.22.1+cu128-cp312-cp312-win_amd64.whl
triton
1.先卸载原有的triton:
python.exe -m pip uninstall triton -y
2安装新的triton:
python.exe -m pip install -U "triton-windows<3.4"
3.将我们提供的python_3.12.7_include_libs文件夹里的所有文件夹复制到你的ComfyUI的环境根目录里,例如 H:\ComfyUI\python_embeded,然后在环境里通过以下代码验证triton是否安装成功
python.exe test.py
如果返回下面相关的代码则正面triton成功安装
tensor([0., 0., 0.], device='cuda:0')
If you see tensor([0., 0., 0.], device='cuda:0'), then it works
SageAttention
1.卸载原有的SageAttention:
python.exe -m pip uninstall sageattention -y
2.下载相匹配的sageattention :Releases · woct0rdho/SageAttention
我们提供的是sageattention-2.1.1+cu128torch2.7.0-cp312-cp312-win_amd64.whl
输入以下代码安装:
python.exe -m pip install H:\Software\Xformers\sageattention-2.1.1+cu128torch2.7.0-cp312-cp312-win_amd64.whl
Xformes
请查看以下链接:ComfyUI正确安装Xformes的方法-数字折叠
其他说明
在升级前请确保已经安装Visual Studio软件以及cuda驱动升级为12.8版本
cuda12.8下载链接:CUDA Toolkit 12.8 Downloads | NVIDIA Developer

暂无评论内容