Fix pl_upgrades enwironment (numpy, cuda, gcc)

2026-06-04 12:44:26 +08:00 · 2024-10-22 19:23:05 +02:00
parent 3bec3e9b2d
commit 6bdbd48fc3
1 changed files with 3 additions and 1 deletions
--- 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