Skip to content

Commit

Permalink
Upper-bound torch version for dev installs only (#2864)
Browse files Browse the repository at this point in the history
  • Loading branch information
brentyi authored Feb 2, 2024
1 parent c5a400c commit 6436742
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -53,8 +53,7 @@ dependencies = [
"scikit-image>=0.19.3",
"splines==0.3.0",
"tensorboard>=2.13.0",
# pin torch to <=2.1 to fix https://github.com/pytorch/pytorch/issues/118736
"torch>=1.13.1,<2.2",
"torch>=1.13.1",
"torchvision>=0.14.1",
"torchmetrics[image]>=1.0.1",
"typing_extensions>=4.4.0",
Expand Down Expand Up @@ -102,6 +101,8 @@ dev = [
# NOTE: Disabling projectaria-tools because it doesn't have prebuilt windows wheels
# Syntax comes from here: https://pip.pypa.io/en/stable/reference/requirement-specifiers/
"projectaria-tools>=1.3.1; sys_platform != 'win32'",
# pin torch to <=2.1 to fix https://github.com/pytorch/pytorch/issues/118736
"torch>=1.13.1,<2.2",
]

# Documentation related packages
Expand Down

0 comments on commit 6436742

Please sign in to comment.