diff --git a/runner-requirements.txt b/runner-requirements.txt index 3f40053..dc9da5c 100644 --- a/runner-requirements.txt +++ b/runner-requirements.txt @@ -6,7 +6,7 @@ # pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 # pip install -r requirements.txt # --------- pytorch --------- # -torch>=1.11.0,<2.0.0 +torch>=1.11.0,<3.0.0 torchvision>=0.11.0 pytorch-lightning==1.8.3 torchmetrics==0.11.0