Commit 99ca7e3a authored by Ulrik Strid's avatar Ulrik Strid
Browse files

ollama: Fix rocm runtime issues

parent dfe95bf2

pkgs/tools/misc/ollama/default.nix

+2 −1
Original line number Diff line number Diff line
@@ -165,7 +165,8 @@ goBuild ((lib.optionalAttrs enableRocm { 
    # expose runtime libraries necessary to use the gpu

    mv "$out/bin/ollama" "$out/bin/.ollama-unwrapped"

    makeWrapper "$out/bin/.ollama-unwrapped" "$out/bin/ollama" \

      --suffix LD_LIBRARY_PATH : '/run/opengl-driver/lib:${lib.makeLibraryPath runtimeLibs}'

      --suffix LD_LIBRARY_PATH : '/run/opengl-driver/lib:${lib.makeLibraryPath runtimeLibs}' '' + lib.optionalString enableRocm ''\

      --set-default HIP_PATH ${pkgs.rocmPackages.meta.rocm-hip-libraries}

  '';



  ldflags = [