diff --git a/requirements.txt b/requirements.txt index 117861a..406c351 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/cu117 -torch==2.0.0+cu117 +torch==2.2.0 torchvision==0.15.1+cu117 torchaudio==2.0.1 opencv-python-headless