diff --git a/test/python/directml/torch/requirements.txt b/test/python/directml/torch/requirements.txt
index e6c3271983..effe87cc0d 100644
--- a/test/python/directml/torch/requirements.txt
+++ b/test/python/directml/torch/requirements.txt
@@ -1,2 +1,2 @@
 --index-url https://download.pytorch.org/whl/cpu
-torch==2.7.1
+torch==2.7.1+cpu