diff --git a/requirements.txt b/requirements.txt index b9ac257..8933676 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ --find-links https://download.pytorch.org/whl/torch_stable.html numpy==1.22.4 opencv-python==4.6.0.66 -torch===1.13.1 +torch===2.2.0 torchvision==0.13.1+cu113 matplotlib==3.5.2 scikit-learn==1.1.2