colin
/
nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #298077 from ulrikstrid/ulrikstrid--rocm-6-ollama-2 

ollama: Fix rocm runtime issues
This commit is contained in:
Pol Dellaiera 2024-03-24 08:09:55 +01:00 committed by GitHub
parent 875d47164e 99ca7e3aa6
commit efbd32472c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pkgs/tools/misc/ollama/default.nix
View File

@ -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 = [