starred/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2026-04-27 11:15:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

update gpu.go

Browse Source
This commit is contained in:
pufferffish
2024-06-15 12:05:01 +01:00
parent e77ea68e11
commit 18f3f960b0
4 changed files with 441 additions and 228 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
gpu/gpu_linux.go
View File

@@ -43,10 +43,26 @@ var OneapiGlobs = []string{
"/usr/lib*/libze_intel_gpu.so*",
}
var VulkanGlobs = []string{
"/usr/lib/x86_64-linux-gnu/libvulkan.so*",
"/usr/lib*/libvulkan.so*",
}
var capLinuxGlobs = []string{
"/usr/lib/x86_64-linux-gnu/libcap.so*",
"/usr/lib*/libcap.so*",
}
var CudartMgmtName = "libcudart.so*"
var NvcudaMgmtName = "libcuda.so*"
var NvmlMgmtName = "" // not currently wired on linux
var OneapiMgmtName = "libze_intel_gpu.so"
var VulkanMgmtName = "libvulkan.so*"
var libcapMgmtName = "libcap.so*"
func FindLibCapLibs() []string {
return FindGPULibs(libcapMgmtName, capLinuxGlobs)
}
func GetCPUMem() (memInfo, error) {
var mem memInfo