diff --git a/requirements.txt b/requirements.txt
index c358a6f..198cb44 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ musdb
 museval
 h5py
 tqdm
-torch==1.4.0
+torch==2.2.0
 torchvision==0.5.0
 tensorboard
 sortedcontainers
\ No newline at end of file