Allow numpy>2 and support compute capability >9

2026-06-04 12:44:26 +08:00 · 2025-04-24 22:25:20 +00:00
parent cb899a5ab4
commit da37880bb7
2 changed files with 2 additions and 1 deletions
--- a/environment.yml
+++ b/environment.yml
@@ -14,7 +14,7 @@ dependencies:
  - pdbfixer
  - pytorch-lightning
  - biopython
-  - numpy<2.0.0
+  - numpy
  - pandas
  - PyYAML
  - requests
--- a/setup.py
+++ b/setup.py
@@ -54,6 +54,7 @@ def get_cuda_bare_metal_version(cuda_dir):
 compute_capabilities = set([
    (5, 2), # Titan X
    (6, 1), # GeForce 1000-series
+    (9, 0), # Hopper
 ])

 compute_capabilities.add((7, 0))