diff --git a/CMakePresets.json b/CMakePresets.json index c789ad7f0..68546bde7 100644 --- a/CMakePresets.json +++ b/CMakePresets.json @@ -28,7 +28,7 @@ "name": "CUDA 12", "inherits": [ "CUDA" ], "cacheVariables": { - "CMAKE_CUDA_ARCHITECTURES": "60;61;62;70;72;75;80;86;87;89;90;90a" + "CMAKE_CUDA_ARCHITECTURES": "50;52;53;60;61;62;70;72;75;80;86;87;89;90;90a" } }, { diff --git a/discover/cuda_common.go b/discover/cuda_common.go index 878cee8cb..048295297 100644 --- a/discover/cuda_common.go +++ b/discover/cuda_common.go @@ -57,7 +57,8 @@ func cudaVariant(gpuInfo CudaGPUInfo) string { } } - if gpuInfo.computeMajor < 6 || gpuInfo.DriverMajor < 12 || (gpuInfo.DriverMajor == 12 && gpuInfo.DriverMinor == 0) { + // driver 12.0 has problems with the cuda v12 library, so run v11 on those older drivers + if gpuInfo.DriverMajor < 12 || (gpuInfo.DriverMajor == 12 && gpuInfo.DriverMinor == 0) { return "v11" } return "v12"