diff --git a/requirements.txt b/requirements.txt index b6d44324..f657011f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ -f https://download.pytorch.org/whl/cu113/torch_stable.html torch==1.11.0+cu113 torchvision==0.12.0+cu113 -pytorch-lightning==1.6.4 +pytorch-lightning==2.4.0 setuptools==70.0.0 torchmetrics==0.8.2