diff --git a/setup-requirements.txt b/setup-requirements.txt
index a715ec0..f719c38 100644
--- a/setup-requirements.txt
+++ b/setup-requirements.txt
@@ -1,7 +1,7 @@
 click>=7.0,<9.0
 click-help-colors>=0.9.1,<0.10
 rich>=11.0,<14.0
-torch==2.1.0+cu121
+torch==2.2.1
 torchaudio==2.1.0+cu121
 torchvision==0.16.0+cu121
 triton==2.1.0
\ No newline at end of file