From a6c53d51a3271fd9b63f8f51c6274a1fcacd3ec9 Mon Sep 17 00:00:00 2001
From: gs-olive <113141689+gs-olive@users.noreply.github.com>
Date: Fri, 22 Sep 2023 15:03:47 -0700
Subject: [PATCH] fix: Update Torchvision version to address dependency
 resolution issue

---
 py/requirements.txt | 2 +-
 pyproject.toml      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/py/requirements.txt b/py/requirements.txt
index 09b9a39aa3..f690e1a030 100644
--- 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
index 27af0da9cd..394dc6a60e 100644
--- 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"