diff --git a/EfficientViT/classification/requirements.txt b/EfficientViT/classification/requirements.txt
index 8e4d4ab2..d382c101 100644
--- a/EfficientViT/classification/requirements.txt
+++ b/EfficientViT/classification/requirements.txt
@@ -1,4 +1,4 @@
-torch==1.11.0 
+torch==2.2.0 
 torchvision
 timm==0.5.4 
 einops==0.4.1