AMD provides wheel files for select Python libraries ensuring optimal performance on AMD Zen CPU processors.
- NumPy with AOCL
- SciPy with AOCL
- PyTorch with AOCL
Highlights of release 5.1 on AMD Zen CPU Architecture
- Enhanced PyTorch performance with AOCL (AMD Optimized CPU Libraries)
- Improved NumPy and SciPy performance with AOCL
- Support for updated Python packages, including::
- PyTorch 2.5.0 and 2.4.0
- NumPy 2.1.3 and 1.26.4
- SciPy 1.14 and 1.13.1
Release Notes
- Ubuntu 22 support added
- AOCL-BLAS and AOCL-LAPACK are compiled with GCC
- Compatible for OS with GLIBC 2.28 or later
- Supported Python versions: Python 3.12 and Python 3.11
Known Issues
- On SciPy, a precision difference has been observed for `TestVectorNorms::test_stable`. The test failed due to SciPy implementation of the assertion rather than an actual error with AOCL-BLAS computation.
Resources and Technical Support
Documentation
Python Libraries with AOCL User Guide
To learn more about the AOCL Data Analytics patch for scikit-learn, visit the AOCL Data-Analytics page.
Support
For support options, refer to Technical Support.
AMD Community
For moderated forums, refer to the AMD community.
Download with End User License Agreement
NumPy | |||||||
File Name | Python Library Version | Python Version | Size | Launch Date | OS | Bitness | Description |
numpy-2.1.3-cp311-cp311-linux_x86_64.whl | 2.1.3 | 3.11 | 21MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled NumPy-2.1.3 with AOCL-5.1 |
numpy-2.1.3-cp312-cp312-linux_x86_64.whl | 2.1.3 | 3.12 | 21MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled NumPy-2.1.3 with AOCL-5.1 |
1.26.4 | 3.11 | 20MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled NumPy-1.26.4 with AOCL-5.1 | |
1.26.4 | 3.12 | 20MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled NumPy-1.26.4 with AOCL-5.1 | |
SciPy | |||||||
File Name | Python Library Version | Python Version | Size | Launch Date | OS | Bitness | Description |
scipy-1.14.1-cp311-cp311-linux_x86_64.whl | 1.14.1 | 3.11 | 41MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled SciPy-1.14.1 with AOCL-5.1 |
scipy-1.14.1-cp312-cp312-linux_x86_64.whl | 1.14.1 | 3.12 | 41MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled SciPy-1.14.1 with AOCL-5.1 |
1.13.1 | 3.11 | 40MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled SciPy-1.13.1 with AOCL-5.1 | |
1.13.1 | 3.12 | 40MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled SciPy-1.13.1 with AOCL-5.1 | |
PyTorch | |||||||
File Name | Python Library Version | Python Version | Size | Launch Date | OS | Bitness | Description |
torch-2.5.0a0-cp311-cp311-linux_x86_64.whl | 2.5.0 | 3.11 | 153MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled Pytorch-2.5.0 with AOCL-5.1 |
torch-2.5.0a0-cp312-cp312-linux_x86_64.whl | 2.5.0 | 3.12 | 153MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled Pytorch2.5.0 with AOCL-5.1 |
2.4.0 | 3.11 | 150MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled Pytorch2.4.0 with AOCL-5.1 | |
2.4.0 | 3.12 | 150MB | 6/04/2025 | Rocky8.x, Rocky 9.x, Ubuntu 22.04 | 64-bit | GCC compiled Pytorch2.4.0 with AOCL-5.1 | |