From 6bdbd48fc36e39623211dbe45457a94f0acc85d8 Mon Sep 17 00:00:00 2001
From: Vaclav Hanzl <64649594+vaclavhanzl@users.noreply.github.com>
Date: Tue, 22 Oct 2024 19:23:05 +0200
Subject: [PATCH] Fix pl_upgrades enwironment (numpy, cuda, gcc)

---
 environment.yml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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