diff --git a/requirements.txt b/requirements.txt
index 25fb98b..61561cc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -181,10 +181,8 @@ terminado==0.18.0
 threadpoolctl==3.2.0
 tinycss2==1.2.1
 tomli==2.0.1
-torch==2.1.1+cu121
-torchaudio==2.1.1+cu121
+torch==2.1.1+cu118
 torchmetrics==1.3.0.post0
-torchvision==0.16.1+cu121
 tornado==6.4
 tqdm==4.66.1
 traitlets==5.14.1