highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-10-10 22:13:22 +02:00
Code Issues Packages Projects Releases Wiki Activity

Let gpu.go and gen_linux.sh also find CUDA on Arch Linux

Browse Source
This commit is contained in:
Alexander F. Rødseth
2024-01-14 13:40:36 +01:00
parent 557110d0ba
commit f4bf1d514f
2 changed files with 27 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
gpu/gpu.go
View File

@@ -39,6 +39,7 @@ var CudaLinuxGlobs = []string{
"/usr/lib/x86_64-linux-gnu/libnvidia-ml.so*",
"/usr/lib/wsl/lib/libnvidia-ml.so*",
"/opt/cuda/lib64/libnvidia-ml.so*",
"/opt/cuda/targets/x86_64-linux/lib/stubs/libnvidia-ml.so*",
"/usr/lib*/libnvidia-ml.so*",
"/usr/local/lib*/libnvidia-ml.so*",
"/usr/lib/aarch64-linux-gnu/nvidia/current/libnvidia-ml.so*",