Update dependency torch to v2.5.1

courtois-neuromod · Oct 29, 2024 · 5f4f118 · 5f4f118
1 parent 117ed67
commit 5f4f118
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -10,7 +10,7 @@ version = "0.0.1"
 # The following assumes that we build on top of the NGC PyTorch Docker image.
 dependencies = [
     # MANDATORY for cneuromax/
-    "torch==2.4.1+cu121", # Tensor manipulation
+    "torch==2.5.1", # Tensor manipulation
     "hydra-core==1.3.2", # Configuration management
     "hydra-zen==0.13.0", # Cleaner Hydra configs
     "beartype==0.19.0", # Dynamic type checking