若使用 PyTorch 2.6 及以上版本,虽然新版本特性吸引人,但升级后可能出现节点启动失败的问题,请注意兼容性风险。
在2.6 版本的Torch中,它对于Weight Only的要求相对比较严格,修正这个问题通常两种方法:
- 改源代码
- 将Torch版本降级到2.5 版本或者以下
我们需要卸载Torch 2.6 和Torchvision、torchAudio、Xformers组件,然后再安装。执行过程如下:
Pip uninstall torch torchvision torch audio xformers
接下来执行如下命令进行Torch 及其他组件安装。
pip install torch==2.5.0 torchvision torchaudio xformers --extra-index https://download.pytorch.org/whl/cu124