diff --git a/Containerfile b/Containerfile index 1aae7aa5..1fe209c5 100644 --- a/Containerfile +++ b/Containerfile @@ -54,6 +54,13 @@ RUN rpm-ostree override remove \ toolbox \ htop +# Install ROCM on non-Nvidia images +RUN if grep -v "nvidia" <<< "${IMAGE_NAME}"; then \ + rpm-ostree install \ + rocm-hip \ + rocm-opencl \ +; fi + # Run firstboot script per-profile RUN mkdir -p "/usr/etc/profile.d/" RUN ln -s "/usr/share/ublue-os/firstboot/launcher/login-profile.sh" \