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
PROGRAMMING --
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 PyTorch.org.
Related News
About The Author
Author picture

Michael Larabel is the principal author of Phoronix.com 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 OpenBenchmarking.org automated benchmarking software. He can be followed via Twitter or contacted via MichaelLarabel.com.

Popular News This Week