Quellcode durchsuchen

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

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.
Daniel Hiltgen vor 8 Monaten
Ursprung
Commit
f29b167e1a
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      gpu/cuda_common.go

+ 1 - 1
gpu/cuda_common.go

@@ -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 "v12"