Allow numpy>2 and support compute capability >9

da37880b · Jennifer Wei · cb899a5a · da37880b · da37880b
Commit da37880b authored Apr 24, 2025 by Jennifer Wei
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

environment.yml environment.yml +1 -1

setup.py setup.py +1 -0

No files found.
--- 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))