diff --git a/spack-environment/packages.yaml b/spack-environment/packages.yaml
index e026404b41886c759bd93a015c33207cde29aaa3..390db996314f092dda41f3f291b15ab765837b95 100644
--- a/spack-environment/packages.yaml
+++ b/spack-environment/packages.yaml
@@ -300,6 +300,9 @@ packages:
   py-toml:
     require:
     - '@0.10.2'
+  py-torch:
+    require:
+    - '@2.0.0'  
   py-uproot:
     require:
     - '@5.0.5'