diff --git a/environment.yml b/environment.yml
index 1fdd0119..403d4697 100644
--- a/environment.yml
+++ b/environment.yml
@@ -18,21 +18,17 @@ dependencies:
   - h5py
   - numpy
   # --------- Deep Learning --------- #
-  # cudatoolkit to specify the cuda driver in the conda env
-  - conda-forge::cudatoolkit=11.3.1 # single equal sign there, not a typo
-  - pytorch::pytorch==1.11.0
-  - pytorch::torchvision==0.12.0
-  - conda-forge::pytorch-lightning==1.5.9
-  - conda-forge::torchmetrics==0.7.*
-  - comet_ml::comet_ml==3.31.*
-  - conda-forge::urllib3<2  # To solve for https://github.com/GeneralMills/pytrends/issues/591
-  - pyg::pytorch-cluster
-  - pyg::pytorch-scatter
-  - pyg::pytorch-sparse
-  - pyg::pyg==2.1.0
+  # --------- Deep Learning --------- #
+  - pytorch=2.1
+  - pytorch-cuda=11.8
+  - torchvision=0.16
+  - lightning=2.0
+  - torchmetrics=0.11
+  - pyg=2.4
+  - comet_ml=3.35
     # Nota: if libcusparse.so.11. errors occur, run
     # export LD_LIBRARY_PATH="/home/${USER}/miniconda/envs/lib:$LD_LIBRARY_PATH"
-    # ou
+    # or
     # export LD_LIBRARY_PATH="/home/${USER}/anaconda3/envs/lib:$LD_LIBRARY_PATH"
     # see https://github.com/pyg-team/pytorch_geometric/issues/2040#issuecomment-766610625
   # --------- geo --------- #
@@ -61,7 +57,6 @@ dependencies:
   - rich # beautiful text formatting in terminal
   - sh # for running bash commands in some tests
   - seaborn # used in some callbacks
-  - pudb # debugger
     # # --------- Documentation --------- #
   - sphinx==4.5.*
   - recommonmark==0.7.*