diff --git a/requirements/datatype_graph.txt b/requirements/datatype_graph.txt index 72c2abdae0..266d3a5cc3 100644 --- a/requirements/datatype_graph.txt +++ b/requirements/datatype_graph.txt @@ -8,5 +8,5 @@ networkx <=3.1 class-resolver >=0.3.2, <=0.4.2 # todo: check if we can bump this versions, ALSO if bumped you need to update CI for find links -torch ==1.13.1 +torch ==2.0.1 torchvision ==0.14.1 diff --git a/requirements/datatype_image_extras.txt b/requirements/datatype_image_extras.txt index a31a6bdf69..39433bfcc2 100644 --- a/requirements/datatype_image_extras.txt +++ b/requirements/datatype_image_extras.txt @@ -13,4 +13,4 @@ fairscale torchmetrics<0.8.0 # effdet had an issue with PL 1.12, and icevision doesn't support effdet's latest version yet (0.3.0) -torch <1.12 +torch <2.1 diff --git a/requirements/testing_serve.txt b/requirements/testing_serve.txt index 99e4817627..59e107ea2f 100644 --- a/requirements/testing_serve.txt +++ b/requirements/testing_serve.txt @@ -7,7 +7,7 @@ # CI: limit the search space for pip/installation sahi ==0.8.19 -torch ==1.10.2 +torch ==2.0.1 torchaudio ==0.10.2 torchvision ==0.11.3 torchmetrics ==0.10.3