diff --git a/spack-environment/packages.yaml b/spack-environment/packages.yaml
index 7626bd60936d3a5be6eefeafe743ead34d0d62ff..a3cf83a9da9768cd10af33fc25c2f7d81e73710e 100644
--- a/spack-environment/packages.yaml
+++ b/spack-environment/packages.yaml
@@ -297,6 +297,9 @@ packages:
   py-toml:
     require:
     - '@0.10.2'
+  py-torch:
+    require:
+    - '@2.0.0'  
   py-uproot:
     require:
     - '@5.0.5'