diff --git a/neural_networks_hero/infer/files/requirements.txt b/neural_networks_hero/infer/files/requirements.txt index 6f749e7..e13acc9 100644 --- a/neural_networks_hero/infer/files/requirements.txt +++ b/neural_networks_hero/infer/files/requirements.txt @@ -1,4 +1,4 @@ numpy==1.24.1 opencv_python==4.7.0.68 Pillow==10.0.1 -torch==1.11.0+cu113 +torch==2.2.0