Intel Extension For Pytorch with Arc A770 and AMD CPU

[zenon]

Does intel_extension_for_pytorch work with the A770 and an AMD CPU?

[kta_intel]

Hi, thanks for posting on our community forums.

Yes, intel_extension_for_pytorch should work on an A770 and AMD CPU station. A couple things to note though:

1. To make use of the A770, you will need to install the intel_extension_for_pytorch + xpu package alongside the necessary GPU drivers and components from the oneAPI base toolkit (see: https://intel.github.io/intel-extension-for-pytorch/xpu/latest/tutorials/installation.html)
2. For operations on the AMD CPU, we don't make any performance claims since it is not part of our validation/support scope so you may not see any optimization benefits

[jitendra42]

In order to save some installation time, you can quickly get started using container, all the required drivers, ITEX & other required dependencies already installed. models/models_v2/pytorch/stable_diffusion/inference/gpu/CONTAINER_FLEX.md at master · IntelAI/models (github.com)