baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use cuda v11 for driver 525 and older (#6620)

Browse source 
It looks like driver 525 (aka, cuda driver 12.0) has problems with the cuda v12 library
we compile against, so run v11 on those older drivers if detected.
This commit is contained in:
Daniel Hiltgen 2024-09-03 17:15:31 -07:00 committed by GitHub
parent 037a4d103e
commit f29b167e1a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
gpu/cuda_common.go
View file

@ -57,7 +57,7 @@ func cudaVariant(gpuInfo CudaGPUInfo) string {
} }
} }
if gpuInfo.computeMajor < 6 || gpuInfo.DriverMajor < 12 { if gpuInfo.computeMajor < 6 || gpuInfo.DriverMajor < 12 || (gpuInfo.DriverMajor == 12 && gpuInfo.DriverMinor == 0) {
return "v11" return "v11"
} }
return "v12" return "v12"