• AMD Support
    • False
    • Hide

      None

      Show
      None
    • False
    • To Do
    • RHOAISTRAT-135 - Support AMD GPUs
    • 85% To Do, 15% In Progress, 0% Done
    • No
    • RHOAI IDE - Boston
    • Testable

      As a User, I want OOTB workbench image(s) with AMD libraries & packages

      Description

      Create the setup for building the AMD-based container image, which supports the jupyter packages and
      AMD ROCm. This is similar to existing CUDA (NVidia) and HabanaAI images.

      Blockers:

      • We need some documentation/github repo to pull the information needed to build the images.
      • Which are the libraries and packages

      Dev:

      • Create the base ROCm Dockerfile with ROCm binaries.
      • Check if there is a separate Tensorflow or PyTorch for AMD.
      • Build Notebook images (Datascience, Tensorflow, PyTorch) with base ROCm image.
      • Test it on an AMD node
      • Find ways to verify AMD binaries are set.
        • `rocminfo` 

       QA:

      • Test it on the AMD node(might require AMD operator)

      Docs:

      • Document the AMD workbench in the RHOAI.

      Questions 

      • Where will be the material we need to fetch to build the images.
        • For NVidia we go to their github and get that information.
      • Security Constraints 

            Unassigned Unassigned
            aparedes@redhat.com Adriel Paredes
            RHOAI IDE
            Votes:
            0 Vote for this issue
            Watchers:
            2 Start watching this issue

              Created:
              Updated: