ollama: Fix rocm runtime issues
This commit is contained in:
parent
dfe95bf220
commit
99ca7e3aa6
|
@ -165,7 +165,8 @@ goBuild ((lib.optionalAttrs enableRocm {
|
||||||
# expose runtime libraries necessary to use the gpu
|
# expose runtime libraries necessary to use the gpu
|
||||||
mv "$out/bin/ollama" "$out/bin/.ollama-unwrapped"
|
mv "$out/bin/ollama" "$out/bin/.ollama-unwrapped"
|
||||||
makeWrapper "$out/bin/.ollama-unwrapped" "$out/bin/ollama" \
|
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 = [
|
ldflags = [
|
||||||
|
|
Loading…
Reference in New Issue
Block a user