search

Akegarasu / lora-scripts

LoRA & Dreambooth training scripts & GUI use kohya-ss's trainer, for diffusion model.
GNU Affero General Public License v3.0
4.39k stars 540 forks source link

使用tagger打标会显示CUDA路径已设置但无法加载CUDA #484

Closed bukkumaaku closed 3 weeks ago

bukkumaaku commented 3 weeks ago

Snipaste_2024-08-25_01-00-45_1 经过排查之后发现是onnxruntime版本不支持最新版cuda所致。 本机（window 11）安装的cuda是12.6，而 ./mikazuki/launch_utils.py 里要求onnxruntime和onnxruntime-gpu的版本是1.17.1，不支持cuda 12.6，但是我尝试降级cuda为11.8也仍然报这个错误。 最后我把 ./mikazuki/launch_utils.py 里的

onnx_version = "1.17.1"

更改为

onnx_version = "1.19.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/"

才可以正常打标