diff --git a/requirements/torch_gpu.txt b/requirements/torch_gpu.txt
index 03e2a6906..1f1abba3f 100644
--- a/requirements/torch_gpu.txt
+++ b/requirements/torch_gpu.txt
@@ -1,4 +1,4 @@
-# Contains packages recommended for functional performance
+# Contains packages requirements for GPU installation
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torch==2.1.1+cu117
---find-links https://data.pyg.org/whl/torch-2.1.0+cu117.html
+torch==2.1.0+cu118
+--find-links https://data.pyg.org/whl/torch-2.1.0+cu118.html