-
Epic
-
Resolution: Unresolved
-
Major
-
None
-
None
-
AMD Support
-
False
-
-
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`
- Reference:
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