diff --git a/docker/requirements.txt b/docker/requirements.txt index 36c97cde..84641db7 100644 --- a/docker/requirements.txt +++ b/docker/requirements.txt @@ -14,7 +14,7 @@ celery[redis,auth,msgpack] #==4.4.7 # for the AI backend opencv-python --find-links https://download.pytorch.org/whl/torch_stable.html -torch==1.9.0+cu111 +torch==2.2.0 torchvision==0.10.0+cu111 # Detectron2 diff --git a/requirements.txt b/requirements.txt index 4daab4cd..2903c719 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,7 +13,7 @@ celery[redis,auth,msgpack] # for the AI backend opencv-python --find-links https://download.pytorch.org/whl/torch_stable.html -torch==1.9.0+cu111 +torch==2.2.0 torchvision==0.10.0+cu111 # Detectron2