PyTorch 1.8 Released With AMD ROCm Binaries

Written by Michael Larabel in Programming on 5 March 2021 at 08:23 AM EST. 26 Comments
PyTorch 1.8 was released on Thursday as the newest version of this widely-used machine learning library. Exciting many will be easier AMD Radeon ROCm support with Python wheels now provided for that Radeon Open eCosystem support.

PyTorch 1.8 brings improvements to distributed training with pipeline parallelism support that will bolster models not fitting on a single GPU device. ability to extend the PyTorch dispatcher for a new back-end in C++, and perhaps most exciting is AMD GPU binaries now being official/

Starting with PyTorch 1.8, AMD ROCm wheels are provided for an easy onboarding process of AMD GPU support for this machine learning library. It's great seeing the recent trend continue of more AI projects and other open-source initiatives finally making official their ROCm support.

PyTorch 1.8 details can be found on
Related News
About The Author
Michael Larabel

Michael Larabel is the principal author of and founded the site in 2004 with a focus on enriching the Linux hardware experience. Michael has written more than 20,000 articles covering the state of Linux hardware support, Linux performance, graphics drivers, and other topics. Michael is also the lead developer of the Phoronix Test Suite, Phoromatic, and automated benchmarking software. He can be followed via Twitter, LinkedIn, or contacted via

Popular News This Week