...
Popular tools such as Pytorch, TensorFlow, and JAX can be used with the Intel distribution for Python (use the offline installer on the login nodes) together with certain special framework-specific extensions. Environments can be separately prepared for each framework below for use with Intel GPUs. Note that the module intel/2024.0.0 (under sw.pvc) must be loaded for these frameworks to be installed or run properly.
We also offer a standalone module (intel_AI_tools/2024/.0.0) that loads a conda installation with the following pre-installed, Intel GPU/XPU-ready environments:
...