Frameworks 2024.1 'Preview' Release with updated Pytorch feature release available on Sunspot
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://urldefense.us/v3/__https://apps.cels.anl.gov/confluence/display/inte... Please let me know of any issues encountered. Rebecca Ramer [email protected]<mailto:[email protected]>
participants (1)
-
Ramer, Rebecca