...
| Hinweis |
|---|
The latest Intel AI tools have specific Intel GPU driver requirements. Currently, only the PVC compute nodes |
Pytorch
Load the Intel OneAPI module and create a new conda environment within your Intel python distribution:
...
Examples of how to use the Intel extension for Pytorch can be found here.
TensorFlow
Similar to Pytorch, an Intel extension for TensorFlow exists. To prepare a TensorFlow environment for use with Intel GPUs, first create a new conda environment:
...
Examples of how to use the Intel extension for TensorFlow can be found here.
JAX
| Hinweis |
|---|
Intel XPU support is still experimental for JAX. |
...