diff --git a/environment.yml b/environment.yml
index 6a8a1c3..4489590 100644
--- a/environment.yml
+++ b/environment.yml
@@ -14,7 +14,7 @@ dependencies:
   - pdbfixer
   - pytorch-lightning
   - biopython
-  - numpy<2.0.0
+  - numpy
   - pandas
   - PyYAML
   - requests
diff --git a/setup.py b/setup.py
index 4873570..bd8c9cf 100644
--- 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))