Researchers and developers working with Machine Learning (ML) models and algorithms using PyTorch can now use AMD ROCm 5.7 on Ubuntu® Linux® to tap into the parallel computing power of the Radeon™ RX 7900 XTX and the Radeon™ PRO W7900 graphics cards which are based on the AMD RDNA™ 3 GPU architecture.
A client solution built on these two high-end GPUs enables a local, private, and cost-effective workflow for ML training and inference for those who previously relied on cloud-based solutions alone.
“We are excited to offer the AI community new support for machine learning development using PyTorch built on the AMD Radeon RX 7900 XTX and Radeon PRO W7900 GPUs and the ROCm open software platform. This is our first RDNA 3 architecture-based implementation, and we are looking forward to partnering with the community.”, says Dan Wood, Corp. Vice President, Radeon Product Management at AMD.
ACCELERATE MACHINE LEARNING ON YOUR DESKTOP
UNIFIED SOFTWARE STACK FOR THE DESKTOP AND THE DATACENTER
FREEDOM TO CUSTOMIZE
As the industry moves towards an ecosystem that supports a broad set of systems, frameworks and accelerators, AMD is determined to continue to make AI more accessible to developers and researchers that benefit from a local client-based setup for ML development using RDNA™ 3 architecture-based desktop GPUs.
AMD is also looking to broaden support for additional ML frameworks and operating systems.
Learn More
https://www.amd.com/en/developer/resources/ml-radeon.html
Download Software
https://www.amd.com/en/support/linux-drivers
Visit the Documentation Portal to get started training ML models on your local desktop
https://rocm.docs.amd.com/projects/radeon/en/latest/
Prerequisites
https://rocm.docs.amd.com/projects/radeon/en/latest/docs/prerequisites.html
How to Guide
https://rocm.docs.amd.com/projects/radeon/en/latest/docs/install/howto.html
© 2023 Advanced Micro Devices, Inc. All rights reserved. AMD, the AMD Arrow logo, CDNA, Radeon, ROCm, and combinations thereof are trademarks of Advanced Micro Devices, Inc. Linux® is the registered trademark of Linus Torvalds in the U.S. and other countries. Microsoft and Windows are registered trademarks of Microsoft Corporation in the US and/or other countries. PyTorch, the PyTorch logo and any related marks are trademarks of The Linux Foundation. TensorFlow, the TensorFlow logo and any related marks are trademarks of Google Inc. Ubuntu and the Ubuntu logo are registered trademarks of Canonical Ltd. Other product names used in this publication are for identification purposes only and may be trademarks of their respective owners.
Radeon™ AI technology is compatible with all AMD Radeon 7000 Series graphics cards and newer. Please check with your system manufacturer for feature availability prior to purchase. GD-232.