ollama - 💡(How to fix) Fix [Linux] Install script doesn't install rocm package when NVIDIA and AMD GPU present [1 participants]

TL;DR

Modify the install script to handle systems with multiple GPUs, including both NVIDIA and AMD, to ensure the rocm package is installed when necessary.

Guidance

• Identify the specific condition in the install script that causes the early exit when an NVIDIA GPU is detected.
• Modify this condition to check for the presence of an AMD GPU as well, and install the rocm package if it is found.
• Consider adding a flag or option to the install script to allow users to force the installation of the rocm package, even if an NVIDIA GPU is present.
• Verify that the modified script correctly handles systems with multiple GPUs and installs the necessary packages.

Example

No specific code snippet can be provided without more information about the install script, but the modification might involve adding an additional check for the AMD GPU, such as:

if [ -n "$NVIDIA_GPU" ] && [ -n "$AMD_GPU" ]; then
  # Install rocm package
fi

Notes

The exact modification will depend on the specifics of the install script and the conditions under which it exits early.

Recommendation

Apply workaround: Modify the install script to correctly handle systems with multiple GPUs, as this will ensure the rocm package is installed when necessary, even if the system has both NVIDIA and AMD GPUs.