diff --git a/deploy/deploy_requirements.txt b/deploy/deploy_requirements.txt index af57c97..5588e29 100644 --- a/deploy/deploy_requirements.txt +++ b/deploy/deploy_requirements.txt @@ -2,4 +2,4 @@ albumentations==1.0.0 matplotlib==3.5.1 numpy==1.22.2 torchvision==0.12.0 -torch==1.11.0 +torch==1.13.1 diff --git a/train/train_requirements.txt b/train/train_requirements.txt index 3dc720e..52cc025 100644 --- a/train/train_requirements.txt +++ b/train/train_requirements.txt @@ -1,5 +1,5 @@ # These versions are pre-installed on the nvidia image, we want to avoid reinstalling -torch==1.11.0a0+17540c5 +torch==1.13.1 torchvision==0.12.0a0 torchtext==0.12.0a0