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

Merge pull request #1874 from dhiltgen/correct_cuda_min

Browse source 
Set corret CUDA minimum compute capability version
This commit is contained in:
Daniel Hiltgen 2024-01-09 11:37:22 -08:00 committed by GitHub
commit e89dc1d54b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
gpu/gpu.go
View file

@ -26,8 +26,8 @@ type handles struct {
var gpuMutex sync.Mutex
var gpuHandles *handles = nil
// TODO verify this is the correct min version
const CudaComputeMajorMin = 5
// With our current CUDA compile flags, 5.2 and older will not work properly
const CudaComputeMajorMin = 6
// Note: gpuMutex must already be held
func initGPUHandles() {