From 2e66cccf50ec5c6949c6d770e3b8013e23ed1f63 Mon Sep 17 00:00:00 2001
From: Victor Mylle <victor.mylle@hotmail.com>
Date: Sun, 16 Apr 2023 10:38:40 +0000
Subject: [PATCH] Updated requirements

---
 requirements.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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