diff --git a/requirements.txt b/requirements.txt
index f339074..2e09c25 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,7 +9,7 @@ opencv-python==4.6.0.66
 plotly==5.11.0
 scikit-learn==1.0.2
 clearml==1.10.3
-torch
-torchvision
+torch==2.0.0+cu118
+torchvision==0.15.1+cu118
 tqdm==4.54.1
 optuna==3.1.1
\ No newline at end of file