Update gpu.go for K80

This commit is contained in:
idream3000
2025-03-05 10:55:31 +08:00
committed by GitHub
parent ea3ec1c13a
commit f32ac1064f

View File

@@ -66,8 +66,8 @@ var (
// With our current CUDA compile flags, older than 5.0 will not work properly
// (string values used to allow ldflags overrides at build time)
var (
CudaComputeMajorMin = "5"
CudaComputeMinorMin = "0"
CudaComputeMajorMin = "3"
CudaComputeMinorMin = "5"
)
var RocmComputeMajorMin = "9"