There is now an updated 2024.1 DL Frameworks release with support for pytorch and tensorflow on the Sunspot deployment. This is a ‘preview’ release which has an update release of Intel pytorch DL Framework, along with the same Intel tensorflow
DL Framework provided in the DL Frameworks deployment announced in April. This deployment uses the ‘preview’ version oneAPI SDK release, Intel's 2024.1.0 oneapi basekit.
This release is available by loading the frameworks modulefile:
module use /soft/preview-modulefiles/24.086.0
module load frameworks/2024.04.15.002
Loading this module activates the aurora_nre_models_frameworks-2024.1_preview_u1 conda environment. Along with the pytorch and tensorflow packages, all necessary python packages needed to run the 11 'Aurora NRE' models have been installed
into this conda environment. An experimental version of Intel's xpu backend for triton is also provided.
More information about this preview release can be found here:
https://apps.cels.anl.gov/confluence/display/inteldga/2024.1+-+Release+Preview
Please let me know of any issues encountered.
Rebecca Ramer