4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00

Upper-bound torch version for dev installs only (#2864)

This commit is contained in:
Brent Yi 2024-02-01 16:19:54 -08:00 committed by GitHub
parent c5a400c8c9
commit 6436742c34
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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",
@ -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