Uploaded image for project: 'AI Platform Core Components'
  1. AI Platform Core Components
  2. AIPCC-7810

Build TensorFlow from source in RHOAI

XMLWordPrintable

    • Icon: Feature Feature
    • Resolution: Unresolved
    • Icon: Normal Normal
    • None
    • None
    • Accelerator Enablement
    • None
    • False
    • Hide

      None

      Show
      None
    • True
    • 40% To Do, 0% In Progress, 60% Done
    • XL
    • Hide

      3/Mar/2026 - awaiting engineer assignment

      CUDA: Green

      ROCm: Red/Green

      For ROCm, we currently ship the wheel as pre-built. Given the difficulties outlined in our attempt to build it from source, it's likely better to just continue shipping the pre-built wheel. The Notebook team has been contacted (link) but has not replied yet per the new pre-built wheel process. Also per the process, we'll need to file a security exception, but the pre-built wheel is already present in builder, so at least the code side for that is done.

       

      Show
      3/Mar/2026 - awaiting engineer assignment CUDA: Green ROCm: Red/Green For ROCm, we currently ship the wheel as pre-built. Given the difficulties outlined in our attempt to build it from source, it's likely better to just continue shipping the pre-built wheel. The Notebook team has been contacted ( link ) but has not replied yet per the new pre-built wheel process. Also per the process, we'll need to file a security exception, but the pre-built wheel is already present in builder, so at least the code side for that is done.  

      ADR: https://docs.google.com/document/d/1YeQHna4dj68GXlx8asPRi_jL05LtT6lmrF6s9moUjLU/edit?usp=sharing  

       

      Feature title:  Build TensorFlow from source

      Feature Overview: TensorFlow is a platform for machine learning initiated and largely maintained by Google. We are currently shipping it as pre built wheels from PyPI, but we'd like to build it from source like other packages. This effort is large as it involves different problems to solve. This initiative will track that effort.

      Product(s) associated:

      RHAIIS: No
      RHEL AI: No
      RHOAI: Yes (notebooks)

      Goals:

      Build accelerator optimized wheels for TensorFlow from source using AIPCC tooling and ecosystem.

      Requirements:

      Builds for CPU, CUDA and ROCm are needed.

      Done - Acceptance Criteria:

      Builds are published in our package index for these variants, and probe tested.

              fjansen@redhat.com Frank Jansen
              emacchi@redhat.com Emilien Macchi
              Frank's Team
              Votes:
              0 Vote for this issue
              Watchers:
              9 Start watching this issue

                Created:
                Updated: