AMD has unveiled an updated ROCm 5.7.1 driver for Ubuntu Linux that brings PyTorch 2.0.1 support for RDNA 3-based Radeon Pro W7900 and Radeon RX 7900 XTX graphics cards. The move enables developers to use AMD's flagship client GPUs with 24GB and 48GB of memory for artificial intelligence applications using PyTorch libraries.
"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.
The driver update is specifically tuned to work with PyTorch 2.0.1, a popular deep learning library, bringing forth capabilities of AMD's RDNA3-based GPUs. The ROCm 5.7.1 driver only supports flagship graphics cards: the Radeon RX 7900 XTX 24GB and the Radeon Pro W7900 48GB. This is logical as AI applications like large language models (LLM) benefit from high-capacity onboard memory, and AMD wants to sell as many high-end boards as possible.
