秋叶Lora训练器升级torch 2.4.1版本方法

11/13/2024, 11:10:13 AM | 热度:17 | 阅读本文需 2 分钟

点赞

image.png

最近都在训练模型所以很少更新教程,因为想到 ComfyUI 把 torch 升级到 2.4.1 后图像推理的速度快了很多,所以想试试把秋叶训练器的torch 也升级一下,看看训练速度是否可以提升一点,顺便把升级的过程分享出来给大家参考

升级方法

建议先备份一下你的 Python 环境,你可以先把 lora-scripts-v1.8.5 目录下的 Python 文件夹打包成一个压缩包。

首先我们需要准备下图的四个文件,放在训练器的 python 路径内

image.png

然后我们在命令行窗口内依次运行以下命令

python.exe -m pip install "torch-2.4.1+cu124-cp310-cp310-win_amd64.whl"

python.exe -m pip install "torchaudio-2.4.1+cu124-cp310-cp310-win_amd64.whl"

python.exe -m pip install "torchvision-0.19.1+cu124-cp310-cp310-win_amd64.whl"

python.exe -m pip install xformers-0.0.28.post1-cp310-cp310-win_amd64.whl

全部显示安装成功就可以了,运行训练器,有显示刚刚安装 torch 版本,然后你随便训练一个模型看看会不会报错,不报错就是成功了

image.png

文件下载

torchvision:https://download.pytorch.org/whl/torchvision/

torch:https://download.pytorch.org/whl/torchvision/

xformers:https://download.pytorch.org/whl/torchvision/

torchaudio:https://download.pytorch.org/whl/torchvision/

这些地址内有很多版本,大家可以复制我前面提供的 pip 安装命令中的版本,再到网页中 ctrl+f 快速定位到需要下载的版本

image.png

模型推荐

最近训练了一堆材质模型,非常适合做材质图标和字体,大家可以去我的主页体验一下,欢迎各位返图

https://www.liblib.art/userpage/dc5bc785ddbb41428f901f7100a45f3c/publish

image.png


0

继续阅读本文相关话题

本文收录于专题

SD教程

共收录篇

查看本专题

文章目录

升级方法

文件下载

模型推荐

AI教程交流群

扫描二维码加入群聊

与同行学习交流&资源共通