diff --git a/gpu/cuda_common.go b/gpu/cuda_common.go index 827cc9b4..aceec70a 100644 --- a/gpu/cuda_common.go +++ b/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"