Add ollama executable peer dir for rocm
This allows people who package up ollama on their own to place the rocm dependencies in a peer directory to the ollama executable much like our windows install flow.
This commit is contained in:
parent
908005d90b
commit
00ec269321
1 changed files with 15 additions and 0 deletions
|
@ -287,6 +287,21 @@ func AMDValidateLibDir() (string, error) {
|
||||||
return rocmTargetDir, nil
|
return rocmTargetDir, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// next to the running binary
|
||||||
|
exe, err := os.Executable()
|
||||||
|
if err == nil {
|
||||||
|
peerDir := filepath.Dir(exe)
|
||||||
|
if rocmLibUsable(peerDir) {
|
||||||
|
slog.Debug("detected ROCM next to ollama executable " + peerDir)
|
||||||
|
return rocmTargetDir, setupLink(peerDir, rocmTargetDir)
|
||||||
|
}
|
||||||
|
peerDir = filepath.Join(filepath.Dir(exe), "rocm")
|
||||||
|
if rocmLibUsable(peerDir) {
|
||||||
|
slog.Debug("detected ROCM next to ollama executable " + peerDir)
|
||||||
|
return rocmTargetDir, setupLink(peerDir, rocmTargetDir)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Well known ollama installer path
|
// Well known ollama installer path
|
||||||
installedRocmDir := "/usr/share/ollama/lib/rocm"
|
installedRocmDir := "/usr/share/ollama/lib/rocm"
|
||||||
if rocmLibUsable(installedRocmDir) {
|
if rocmLibUsable(installedRocmDir) {
|
||||||
|
|
Loading…
Reference in a new issue