这阵子 Flux Kontext 的曝光度日益见长，忍不住想体验一下，但 Kontext 要求 PyTorch 版本在 2.5 以上。我打开自己的绘世启动器，看了一下 PyTorch 的版本，还是 2.1.2+cu118，差得有点多。尝试用绘世启动器内置的环境维护升级 PyTorch，不知为何一直不成功：
看报错是网络问题，但开了翻墙依然如此，只好手动处理之。
首先我们要确认自己能用的 CUDA 版本，以本人的 Windows 11 为例，打开终端，输入命令：
nvidia-smi
我的显卡是 4090，当前显卡驱动能安装的 CUDA 版本是 12.9 。但考虑到 PyTorch 还没有兼容 CUDA 12.9，我们转到 NVIDIA 的官方网站，下载 12.8 就好了。
安装好 CUDA 12.8 之后，我们选择手动下载 PyTorch 的相关文件。打开 https://download.pytorch.org/whl/cu128 ，分别打开其中的 torch / torchaudio / torchvision / xformers，找到我们要下载的文件。
如果您的配置和我一样，CUDA 版本是 12.8，Python 版本是 3.10，那可以直接点击下载下面的四个文件。如果配置和版本不一样，那就只能自己找对应的版本了，建议直接以关键词 cu128 来搜索。
- torch-2.7.0+cu128-cp310-cp310-win_amd64.whl
- torchaudio-2.7.0+cu128-cp310-cp310-win_amd64.whl
- torchvision-0.22.0+cu128-cp310-cp310-win_amd64.whl
- xformers-0.0.30-cp310-cp310-win_amd64.whl
Torch 和其他三个包的对应关系请参考下面的表格，一定要严格对应，非对应版本无法安装。
| Cuda | torch | torchaudio | torchvision | xformers | Python |
|---|---|---|---|---|---|
| cu118, cu126, cu128 | 2.7.0 | 2.7.0 | 0.22.0 | v0.0.30 | >=3.9, <=3.12 |
| cu118, cu124, cu126 | 2.6.0 | 2.6.0 | 0.21.0 | v0.0.29.post3, v0.0.29.post2 | >=3.9, <=3.12 |
| cu118, cu124, cu126 | 2.5.1 | 2.5.1 | 0.20.1 | 0.0.29.post1, 0.0.29, 0.0.28.post3 | >=3.9, <=3.12 |
| cu118, cu121, cu124 | 2.5.0 | 2.5.0 | 0.20.0 | 0.0.28.post2 | >=3.9, <=3.12 |
| cu118, cu121, cu124 | 2.4.1 | 2.4.1 | 0.19.1 | 0.0.28.post1 | >=3.9, <=3.12 |
| cu118, cu121, cu124 | 2.4.0 | 2.4.0 | 0.19.0 | 0.0.27.post2 | >=3.8, <=3.12 |
| cu118, cu121 | 2.3.1 | 2.3.1 | 0.18.1 | 0.0.27.post1 | >=3.8, <=3.12 |
| cu118, cu121 | 2.3.0 | 2.3.0 | 0.18.0 | 0.0.27 | >=3.8, <=3.12 |
| cu118, cu121 | 2.2.2 | 2.2.2 | 0.17.2 | 0.0.26 | >=3.8, <=3.11 |
| cu118, cu121 | 2.2.1 | 2.2.1 | 0.17.1 | 0.0.25 | >=3.8, <=3.11 |
| cu118, cu121 | 2.2.0 | 2.2.0 | 0.17.0 | 0.0.24 | >=3.8, <=3.11 |
| cu118, cu121 | 2.1.2 | 2.1.2 | 0.16.2 | 0.0.23 | >=3.8, <=3.11 |
| cu118, cu121 | 2.1.1 | 2.1.1 | 0.16.1 | 0.0.23 | >=3.8, <=3.11 |
| cu118, cu121 | 2.1.0 | 2.1.0 | 0.16.0 | 0.0.22.post7 | >=3.8, <=3.11 |
| 2.0.1 | 2.0.2 | 0.15.2 | 0.0.22, 0.0.21, 0.0.20 | >=3.8, <=3.11 | |
| 2.0.0 | 2.0.1 | 0.15.1 | 0.0.18, 0.0.19 | >=3.8, <=3.11 |
把这四个文件放到 ComfyUI 启动器根目录下面的 python 文件夹，然后打开启动器，点「高级选项」菜单，打开右上角的「启动命令提示符」：
我们可以先确认一下当前的 PyTorch 版本：
pip show torch然后卸载之：
pip uninstall -y torch torchaudio torchvision xformers最后依次运行以下命令：
pip install "torch-2.7.0+cu128-cp310-cp310-win_amd64.whl"
pip install "torchaudio-2.7.0+cu128-cp310-cp310-win_amd64.whl"
pip install "torchvision-0.22.0+cu128-cp310-cp310-win_amd64.whl"
pip install "xformers-0.0.30-cp310-cp310-win_amd64.whl"不同版本文件名不同，请别忘了自行替换。安装完之后可以再确认一下 PyTorch 版本，然后进入 ComfyUI 看看是不是能跑 Kontext 了？
