Переглянути джерело

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 8 місяців тому
батько
коміт
f29b167e1a
1 змінених файлів з 1 додано та 1 видалено
  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"