diff --git a/environment.yml b/environment.yml
index 41e4202e5..ad28bf507 100644
--- a/environment.yml
+++ b/environment.yml
@@ -5,6 +5,8 @@ channels:
   - pytorch
   - nvidia 
 dependencies:
+  - cuda
+  - gcc=12.4
   - python=3.10
   - libgcc=7.2
   - setuptools=59.5.0
@@ -13,7 +15,7 @@ dependencies:
   - pdbfixer
   - pytorch-lightning
   - biopython
-  - numpy
+  - numpy<2.0.0
   - pandas
   - PyYAML==5.4.1
   - requests