Update setup.py

NVlabs · Jan 23, 2025 · af1703e · af1703e
1 parent 2ec3476
commit af1703e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bindings/torch/setup.py b/bindings/torch/setup.py
@@ -28,7 +28,7 @@ def max_supported_compute_capability(cuda_version):
 		return 80
 	elif cuda_version < parse_version("11.8"):
 		return 86
-	elif cuda_version < parse_version("12.6"):
+	elif cuda_version < parse_version("12.8"):
 		return 90
 	else:
 		return 120