diff --git a/neptune-database-graphstorm-online-inference/requirements-ndb.txt b/neptune-database-graphstorm-online-inference/requirements-ndb.txt
index 7e4ab5b..7ac88ab 100644
--- a/neptune-database-graphstorm-online-inference/requirements-ndb.txt
+++ b/neptune-database-graphstorm-online-inference/requirements-ndb.txt
@@ -1,7 +1,7 @@
 --index-url https://download.pytorch.org/whl/cpu
 --extra-index-url https://pypi.org/simple
 
-torch==2.1.0
+torch==2.7.1+cpu
 
 -f https://data.dgl.ai/wheels-internal/repo.html
 dgl==1.1.3