fix: Update Torchvision version to address dependency resolution issue (

#2339)
pytorch · Sep 25, 2023 · ecdc040 · ecdc040
1 parent 670d2be
commit ecdc040
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/py/requirements.txt b/py/requirements.txt
@@ -3,7 +3,7 @@ packaging
 pybind11==2.6.2
 --extra-index-url https://download.pytorch.org/whl/nightly/cu121
 torch>=2.2.0.dev,<2.3.0
-torchvision>=0.16.0.dev,<0.17.0
+torchvision>=0.17.0.dev,<0.18.0
 --extra-index-url https://pypi.ngc.nvidia.com
 tensorrt==8.6.1
 pyyaml
diff --git a/pyproject.toml b/pyproject.toml
@@ -52,7 +52,7 @@ dependencies = [
 dynamic = ["version"]
 
 [project.optional-dependencies]
-torchvision = ["torchvision >=0.16.dev,<0.17.0"]
+torchvision = ["torchvision >=0.17.dev,<0.18.0"]
 
 [project.urls]
 Homepage = "https://pytorch.org/tensorrt"