【2026.04.30更新】Deepfacelab 猫之汉化pytorch版

wtxx8888

haiou_f 发表于 2026-5-4 23:24
感谢您的指导！
我刚安装了最新版的驱动，并且显示cuda版本为12.6，但提取时还是转到cpu，我看英文里的说 ...

hzmoopp

haiou_f 发表于 2026-5-4 23:24
感谢您的指导！
我刚安装了最新版的驱动，并且显示cuda版本为12.6，但提取时还是转到cpu，我看英文里的说 ...

我的1060laptop改完可以了，但是另一个电脑的directml一直用不了


@echo .

call setenv.bat


cd .\python


python -m pip install --upgrade pip

python -m pip uninstall torch -y

python -m pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu118

pauss

wtxx8888

hzmoopp 发表于 2026-5-5 14:31
我的1060laptop改完可以了，但是另一个电脑的directml一直用不了

DML用的是  tensorflow-directml  和pytorch-directml，  不用指定torch等 主体的版本。
你百度好directml 的版本后（根据python解释器版本），直接装directml即可。
directml 会自动拉取，匹配的torch或tensorflow。（装之前directml之前，卸载TF跟torch 主体）
完整DX12，最低需要GTX960 及同级显卡。（再低的N卡，可以尝试用CUDA版，装更低版本的CUDA 及torch，比如CUDA 9.2）