diff --git a/SegFormer/requirements-cpu.txt b/SegFormer/requirements-cpu.txt index f44500a..b8274df 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.1 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..a42e3be 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.1 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..94e9283 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.1 ray[serve]==2.0.0 fastapi==0.81.0 imageio==2.21.2