mirror of
https://github.com/ollama/ollama.git
synced 2025-10-06 20:44:37 +02:00
Set corret CUDA minimum compute capability version
If you attempt to run the current CUDA build on compute capability 5.2 cards, you'll hit the following failure: cuBLAS error 15 at ggml-cuda.cu:7956: the requested functionality is not supported
This commit is contained in:
@@ -26,8 +26,8 @@ type handles struct {
|
|||||||
var gpuMutex sync.Mutex
|
var gpuMutex sync.Mutex
|
||||||
var gpuHandles *handles = nil
|
var gpuHandles *handles = nil
|
||||||
|
|
||||||
// TODO verify this is the correct min version
|
// With our current CUDA compile flags, 5.2 and older will not work properly
|
||||||
const CudaComputeMajorMin = 5
|
const CudaComputeMajorMin = 6
|
||||||
|
|
||||||
// Note: gpuMutex must already be held
|
// Note: gpuMutex must already be held
|
||||||
func initGPUHandles() {
|
func initGPUHandles() {
|
||||||
|
Reference in New Issue
Block a user