diff --git a/SegFormer/requirements-cpu.txt b/SegFormer/requirements-cpu.txt index f44500a..34c48e7 100644 --- a/SegFormer/requirements-cpu.txt +++ b/SegFormer/requirements-cpu.txt @@ -1,5 +1,5 @@ torch==1.7.1+cpu -torchvision==0.8.2+cpu +torchvision==0.14.0 mmcv-full==1.2.7 timm==0.3.2 pycocotools==2.0.4 diff --git a/requirements-cpu.txt b/requirements-cpu.txt index e503987..d2138c1 100644 --- a/requirements-cpu.txt +++ b/requirements-cpu.txt @@ -1,5 +1,5 @@ torch==1.12.0+cpu -torchvision==0.13.0+cpu +torchvision==0.14.0 ray[serve]==2.0.0 fastapi==0.81.0 imageio==2.21.2 diff --git a/requirements-gpu.txt b/requirements-gpu.txt index 492895c..ac1ab5e 100644 --- a/requirements-gpu.txt +++ b/requirements-gpu.txt @@ -1,5 +1,5 @@ torch==1.12.0+cu116 -torchvision==0.13.0+cu116 +torchvision==0.14.0 ray[serve]==2.0.0 fastapi==0.81.0 imageio==2.21.2