diff --git a/spack-environment/packages.yaml b/spack-environment/packages.yaml
index 5cd62a81139968afe1b8ee8be2d3c528c21d9012..f97c5724da6d6f615bb8140754044bf08bde34de 100644
--- a/spack-environment/packages.yaml
+++ b/spack-environment/packages.yaml
@@ -284,6 +284,9 @@ packages:
   py-toml:
     require:
     - '@0.10.2'
+  py-torch:
+    require:
+    - '@2.0.0'  
   py-uproot:
     require:
     - '@5.0.5'