PyTorch 2.4
This document represents a proposed Change. As part of the Changes process, proposals are publicly announced in order to receive community feedback. This proposal will only be implemented if approved by the Fedora Engineering Steering Committee.
Summary
PyTorch is a popular python library for deep learning using GPUs and CPUs. It is frequently associated with various AI implementations and other neural network based projects.
Owner
- Name: Tom Rix
- Email: trix@redhat.com
Current status
- Targeted release: Fedora Linux 41
- Last updated: 2024-07-17
- [Announced]
- [<will be assigned by the Wrangler> Discussion thread]
- FESCo issue: <will be assigned by the Wrangler>
- Tracker bug: <will be assigned by the Wrangler>
- Release notes tracker: <will be assigned by the Wrangler>
Detailed Description
The goal for this change is to make sure that the latest pytorch is available in the Fedora repositories. The latest version at the time of F41's release will be 2.4 which will improve AMD ROCm support
Feedback
Benefit to Fedora
Fedora will benefit from this change in multiple ways:
- Having the latest release of PyTorch aids the "First" teneant of Fedora's 4 Fs
- Better integration with AMD's ROCm will make the package more performant and useful
Scope
- Proposal owners: Package and test PyTorch 2.4 with ROCm acceleration enabled
- Other developers:
- Release engineering: N/A (not needed for this Change)
- Policies and guidelines: N/A (not needed for this Change)
- Trademark approval: N/A (not needed for this Change)
- Alignment with the Fedora Strategy: "Innovation & Leadership in Technology: We integrate programming language stack ecosystems" by including the latest release of PyTorch, Fedora will have the latest release of a popular software stack in AI/ML.
Upgrade/compatibility impact
There should be no backwards incompatible changes with the 2.4 release.
Early Testing (Optional)
Do you require 'QA Blueprint' support? N
How To Test
Testing can be done with CPU only but that may lead to very long test times. Testing with ROCm compatible hardware is recommended.
Running through some of the PyTorch Examples is a good way for basic testing of PyTorch.
User Experience
Users will be able to continue to use PyTorch installed from the Fedora repos - the upgrades to llvm require a newer ROCm and that newer ROCm is only supported by the latest PyTorch release.
Other than being able to continue to use PyTorch from the Fedora repos, the user-noticeable changes will be minmal.
Dependencies
There are no RPMs affected by this change which are not included in the change.
Contingency Plan
- Contingency mechanism: (What to do? Who will do it?) N/A (not a System Wide Change)
- Contingency deadline: N/A (not a System Wide Change)
- Blocks release? N/A (not a System Wide Change), Yes/No
If the feature is not complete before final freeze, we will attempt to get the older PyTorch 2.3 working with ROCm 6.2.
Documentation
N/A (not a System Wide Change)