diff --git a/gpu/gpu.go b/gpu/gpu.go index f546506b..781e23df 100644 --- a/gpu/gpu.go +++ b/gpu/gpu.go @@ -31,8 +31,8 @@ type handles struct { } const ( - cudaMinimumMemory = 256 * format.MebiByte - rocmMinimumMemory = 256 * format.MebiByte + cudaMinimumMemory = 457 * format.MebiByte + rocmMinimumMemory = 457 * format.MebiByte ) var gpuMutex sync.Mutex diff --git a/gpu/gpu_darwin.go b/gpu/gpu_darwin.go index 0ba02e1b..f8cc1adb 100644 --- a/gpu/gpu_darwin.go +++ b/gpu/gpu_darwin.go @@ -15,7 +15,7 @@ import ( ) const ( - metalMinimumMemory = 384 * format.MebiByte + metalMinimumMemory = 512 * format.MebiByte ) func GetGPUInfo() GpuInfoList {