ComfyUI SDWebUI升级pytorch随记

原创

已于 2025-05-03 14:07:05 修改 · 2.8k 阅读

于 2024-03-27 00:11:58 首次发布

博主将torch从1.6版本升级到2.1.1版本，升级过程占用近3G空间且报错，提示xformers需升级，又升级了xformers并安装pyre - extensions。拉取最新ComfyUI能正常运行，但SDWebUI 1.6启动报错，修改文件调用后启动成功，还尝试用参数开启ComfyUI的fp8精度。

目前使用的版本是去年10月的1.6版本，有点老。希望支持新的特性，于是乎开始作死。从升级torch开始。先看看已有的版本：

(venv) root@ubuntu-sd-server:~# pip show torch
Name: torch
Version: 2.0.1
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Home-page: https://pytorch.org/
Author: PyTorch Team
Author-email: packages@pytorch.org
License: BSD-3
Location: /root/sd-webui-aki-v4.4/venv/lib/python3.10/site-packages
Requires: filelock, jinja2, networkx, nvidia-cublas-cu11, nvidia-cuda-cupti-cu11, nvidia-cuda-nvrtc-cu11, nvidia-cuda-runtime-cu11, nvidia-cudnn-cu11, nvidia-cufft-cu11, nvidia-curand-cu11, nvidia-cusolver-cu11, nvidia-cusparse-cu11, nvidia-nccl-cu11, nvidia-nvtx-cu11, sympy, triton, typing-extensions
Required-by: accelerate, basicsr, clean-fid, clip, controlnet-aux, facexlib, gfpgan, invisible-watermark, kornia, lpips, open-clip-torch, pytorch-lightning, realesrgan, thop, timm, tomesd, torchdiffeq, torchmetrics, torchsde, torchvision, triton, ultralytics, xformers

然后开始升级最新的2.1.1。幸好空间留的大，升级把一堆相关的都升上去了，好家伙，一下干了快3个G。

(venv) root@ubuntu-sd-server:~# pip install --upgrade torch torchvision
Looking in indexes: https://pypi.doubanio.com/simple
Requirement already satisfied: torch in ./sd-webui-aki-v4.4/venv/lib/python3.10/site-packages (2.0.1)
Collecting torch
  Downloading https://mirrors.cloud.tencent.com/pypi/packages/a7/ad/fbe7d4cffb76da4e478438853b51305361c719cff929ab70a808e7fb75e7/torch-2.2.1-cp310-cp310-manylinux1_x86_64.whl (755.5 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 755.5/755.5 MB 735.0 kB/s eta 0:00:00
Requirement already satisfied: torchvision in ./sd-webui-aki-v4.4/venv/lib/python3.10/site-packages (0.15.2)
Collecting torchvision
  Downloading https://mirrors.cloud.tencent.com/pypi/packages/1a/e8/13432ae6be567b577a4c89d1bd50084e4d989b379a7be8050380b5ab3a6e/torchvision-0.17.1-cp310-cp310-manylinux1_x86_64.whl (6.9 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 6.9/6.9 MB 4.2 MB/s eta 0:00:00
Requirement already satisfied: filelock in ./sd-webui-aki-v4.4/venv/lib/python3.10/site-packages (from torch) (3.12.4)
Requirement already satisfied: typing-extensions>=4.8.0 in ./sd-webui-aki-v4.4/venv/lib/python3.10/site-packages (from torch) (4.8.0)
Requirement already satisfied: sympy in ./sd-webui-aki-v4.4/venv/lib/python3.10/site-packages (from torch) (1.12)
Requirement already satisfied: networkx in ./sd-webui-aki-v4.4/venv/lib/python3.10/site-packages (from torch) (3.1)
Requirement already satisfied: jinja2 in ./sd-webui-aki-v4.4/venv/lib/python3.10/site-packages (from torch) (3.1.2)
Requirement already satisfied: fsspec in ./sd-webui-aki-v4.4/venv/lib/python3.10/site-packages (from torch) (2023.9.2)
Collecting nvidia-cuda-nvrtc-cu12==12.1.105 (from torch)
  Downloading https://mirrors.cloud.tencent.com/pypi/packages/b6/9f/c64c03f49d6fbc56196664d05dba14e3a561038a81a638eeb47f4d4cfd48/nvidia_cuda_nvrtc_cu12-12.1.105-py3-none-manylinux1_x86_64.whl (23.7 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 23.7/23.7 MB 5.5 MB/s eta 0:00:00
Collecting nvidia-cuda-runtime-cu12==12.1.105 (from torch)
  Downloading https://mirrors.cloud.tencent.com/pypi/packages/eb/d5/c68b1d2cdfcc59e72e8a5949a37ddb22ae6cade80cd4a57a84d4c8b55472/nvidia_cuda_runtime_cu12-12.1.105-py3-none-manylinux1_x86_64.whl (823 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 823.6/823.6 kB 6.4 MB/s eta 0:00:00
Collecting nvidia-cuda-cupti-cu12==12.1.105 (from torch)
  Downloading https://mirrors.cloud.tencent.com/pypi/packages/7e/00/6b218edd739ecfc60524e585ba8e6b00554dd908de2c9c66c1af3e44e18d/nvidia_cuda_cupti_cu12-12.1.105-py3-none-manylinux1_x86_64.whl (14.1 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 14.1/14.1 MB 5.8 MB/s eta 0:00:00
Collecting nvidia-cudnn-cu12==8.9.2.26 (from torch)
  Downloading https://mirrors.cloud.tencent.com/pypi/packages/ff/74/a2e2be7fb83aaedec84f391f082cf765dfb635e7caa9b49065f73e4835d8/nvidia_cudnn_cu12-8.9.2.26-py3-none-manylinux1_x86_64.whl (731.7 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 731.7/731.7 MB 825.3 kB/s eta 0:00:00
Collecting nvidia-cublas-cu12==12.1.3.1 (from torch)
  Downloading https://mirrors.cloud.tencent.com/pypi/pa

标签