diff --git a/requirements.txt b/requirements.txt index c786867..9085092 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ pandas==0.25.3 nilearn==0.6.0 scikit-learn==0.22.1 ipython -torch==1.5.1 +torch==2.2.0 torchvision==0.6.1 tensorboardX==2.1 tqdm