diff --git a/gpu/gpu.go b/gpu/gpu.go
index cf2f3b7f..708fad10 100644
--- a/gpu/gpu.go
+++ b/gpu/gpu.go
@@ -30,8 +30,8 @@ type handles struct {
 }
 
 const (
-	cudaMinimumMemory = 377 * format.MebiByte
-	rocmMinimumMemory = 377 * format.MebiByte
+	cudaMinimumMemory = 457 * format.MebiByte
+	rocmMinimumMemory = 457 * format.MebiByte
 )
 
 var gpuMutex sync.Mutex