Intel® Optimized AI Frameworks
Receive community support for questions related to PyTorch* and TensorFlow* frameworks.
80 Discussions

Intel Extension For Pytorch with Arc A770 and AMD CPU

zenon
Novice
2,316 Views

Does intel_extension_for_pytorch work with the A770 and an AMD CPU?

Labels (1)
0 Kudos
2 Replies
kta_intel
Moderator
2,260 Views

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
0 Kudos
jitendra42
Employee
1,774 Views

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) 

0 Kudos
Reply