TL;DR

Remove the exit 0 calls in the NVIDIA and AMD GPU checks to allow for proper installation of CUDA and ROCm support in mixed GPU setups.

Guidance

• Identify the lines with exit 0 in the installer script and remove them to prevent premature termination.
• Verify that the installer correctly detects and installs support for both NVIDIA and AMD GPUs after modifying the script.
• Test the installer in different scenarios, including single-card and mixed-card setups, to ensure proper functionality.
• Consider adding additional logging or debugging statements to monitor the installer's progress and detect potential issues.

Example

# Modified code snippet
if check_gpu nvidia-smi; then  
    status "NVIDIA GPU installed."
    # Remove exit 0 to continue with the installation process
fi

# ...

if check_gpu lspci amdgpu || check_gpu lshw amdgpu; then
    download_and_extract "https://ollama.com/download" "$OLLAMA_INSTALL_DIR" "ollama-linux-${ARCH}-rocm"   
    install_success
    status "AMD GPU ready."
    # Remove exit 0 to allow for CUDA installation in mixed GPU setups
fi

Notes

The provided solution assumes that removing the exit 0 calls will allow the installer to correctly detect and install support for both NVIDIA and AMD GPUs. However, additional testing and verification are necessary to ensure the installer functions as expected in different scenarios.

Recommendation

Apply the workaround by removing the exit 0 calls in the NVIDIA and AMD GPU checks, as this will allow the installer to continue with the installation process and potentially fix the issue.