diff options
author | Pol Dellaiera <pol.dellaiera@protonmail.com> | 2024-03-24 15:31:10 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-03-24 15:31:10 +0100 |
commit | 2e1b779e0a0e67fb4e4f3333606cc1b8a287afae (patch) | |
tree | f0ab951e7cdd62e8e31323d724590b885ba9fb48 | |
parent | 4baf0ed6f66c2109965d036d17cf321247b99650 (diff) | |
parent | ea8ab5ba8ee6696d61dbd0aef9241e636b9aca62 (diff) |
Merge pull request #298575 from abysssol/ollama-tidy
ollama: fix a symlink and use a smaller rocm path
-rw-r--r-- | pkgs/tools/misc/ollama/default.nix | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/pkgs/tools/misc/ollama/default.nix b/pkgs/tools/misc/ollama/default.nix index 6d7f2241fbe8e..fc3320d9e0a5b 100644 --- a/pkgs/tools/misc/ollama/default.nix +++ b/pkgs/tools/misc/ollama/default.nix @@ -68,6 +68,10 @@ let cudaPackages.cudatoolkit cudaPackages.cuda_cudart ]; + postBuild = '' + rm "$out/lib64" + ln -s "lib" "$out/lib64" + ''; }; runtimeLibs = lib.optionals enableRocm [ @@ -166,7 +170,7 @@ goBuild ((lib.optionalAttrs enableRocm { 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}' '' + lib.optionalString enableRocm ''\ - --set-default HIP_PATH ${pkgs.rocmPackages.meta.rocm-hip-libraries} + --set-default HIP_PATH ${rocmPath} ''; ldflags = [ |