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

[opendatahub-io/base-containers] ROCm base image should configure ROCm package index for tensorflow-rocm

XMLWordPrintable

    • False
    • Hide

      None

      Show
      None
    • False

      [4031432345] Upstream Reporter: Lalatendu Mohanty
      Upstream issue status: Open
      Upstream description:

      Notebooks declares `tensorflow-rocm` as a direct URL dependency pointing at a specific wheel on `repo.radeon.com`, locked to a single ROCm version (6.4), architecture (x86_64), and Python version (cp312). This cannot work on AIPCC unchanged.

      ```
      tensorflow-rocm @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4/tensorflow_rocm-2.18.1-cp312-cp312-manylinux_2_28_x86_64.whl
      ```

      Base-containers ROCm images (when added) should set `UV_EXTRA_INDEX_URL` or equivalent to point at the ROCm package repository, so consumers can declare `tensorflow-rocm~=2.18.0` as a normal version-pinned dependency and have it resolve from the index configured by the base image.

      *What to do:*

      • Configure ROCm base images with the ROCm package index URL via environment variable
      • Consumers replace direct URL dependencies with plain version pins

      *Affected consumer files:*

      • `jupyter/rocm/tensorflow/ubi9-python-3.12/pyproject.toml`
      • `runtimes/rocm-tensorflow/ubi9-python-3.12/pyproject.toml`

      Parent: #77

              Unassigned Unassigned
              upstream-sync Upstream Sync
              Antonio's Team
              Votes:
              0 Vote for this issue
              Watchers:
              1 Start watching this issue

                Created:
                Updated: