Merge pull request #1797 from AdeelH/pytorch-2.0

Bump torch to version 2.0
azavea · Jun 14, 2023 · 4a4e709 · 4a4e709
2 parents 5ebc65c + c0c57f0
commit 4a4e709
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rastervision_pytorch_learner/requirements.txt b/rastervision_pytorch_learner/requirements.txt
@@ -2,8 +2,8 @@ rastervision_pipeline==0.20.3-dev
 rastervision_core==0.20.3-dev
 numpy==1.23.3
 pillow==9.3.0
-torch==1.12.1
-torchvision==0.13.1
+torch==2.0.1
+torchvision==0.15.2
 tensorboard==2.13.0
 albumentations==1.3.0
 cython==0.29.32