Fix the issue where CUDA version less than 12. x may cause running on CPU #111
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
When I run this node in the CUDA11.8 environment, there is a situation where the CPU is running at full load while the GPU is not, resulting in extremely slow running speed of the video face swapping workflow.
I found that the installation of onnxrruntime-gpu 1.12.1, which is not compatible with the current environment, was caused by running the install.by script.
elif not is_installed(ort,"1.16.1",False): run_pip(ort, "-U")After I specified the installation of version 1.16.1, I could clearly feel that the GPU was involved in the computation
run_pip(ort, "onnxruntime-gpu==1.16.1")By the way, the nsfw check also takes a lot of time due to the default use of CPU.
predict = pipeline("image-classification", model=model_path)