瀏覽代碼

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"