diff --git a/requirements.txt b/requirements.txt
index e277e6ab..09d32907 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cu113
-torch==1.12.1
-torchvision==0.13.1
+torch==2.0.1
+torchvision==0.15.2
 albumentations==0.4.3
 opencv-python==4.5.5.64
 pudb==2019.2