1 changed files with 2 additions and 2 deletions
--- 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