Skip to content

Conversation

@WizardlyBump17
Copy link

@WizardlyBump17 WizardlyBump17 commented Jul 25, 2025

This Pull Request upgrades the Pytorch related stuff to their latest versions for XPU. In order for it to work, the user must have intel-oneapi-base-toolkit-2025.0 installed (it would be nice if there was a warning for it on setup.sh, but I will leave that to you).

It also upgrades the OneAPI and ONNX related stuff. After some testing, I found out that the onnxruntime is needed for the WD14 Captioning to work properly.

I also made a Dockerfile for it. It currently uses my fork while this Pull Request is not merged: https://gist.github.com/WizardlyBump17/3aebf0e1731500e8f4afcba22055b434.

This Pull Request partially fixes #3333, as the correct fix would be to make it work with intel-oneapi-base-toolkit, but that is only possible on nightly versions of Pytorch for XPU, but intel-extension-for-pytorch requires the stable 2.7.1 version.

@bmaltais bmaltais merged commit 80d3b02 into bmaltais:dev Jul 26, 2025
@WizardlyBump17
Copy link
Author

After some investigation, it looks like the intel-oneapi-base-toolkit is not needed at all. When running source /opt/intel/oneapi/setvars.sh, it seems to "poison" some libraries with different versions.

@bmaltais
Copy link
Owner

Does the PR need updating to undo some of the changes?

@WizardlyBump17
Copy link
Author

Does the PR need updating to undo some of the changes?

It doesnt. The only changes was to add onnxruntime and to upgrade Pytorch, which, according to some post I couldnt recall, increases the performance compared to older versions. I felt a performance increase from 2.7.0 to 2.8.0 nightly

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants