From c60bab4d7f98c0be8b81375ee383f8b7ded8ae13 Mon Sep 17 00:00:00 2001 From: gs-olive <113141689+gs-olive@users.noreply.github.com> Date: Fri, 19 Apr 2024 10:48:54 -0700 Subject: [PATCH] fix: Bump `torchvision` version --- py/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/requirements.txt b/py/requirements.txt index 6dce409fe4..3571b9ac34 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.4.0.dev,<2.5.0 -torchvision>=0.18.0.dev,<0.19.0 +torchvision>=0.19.0.dev,<0.20.0 --extra-index-url https://pypi.ngc.nvidia.com tensorrt==8.6.1 pyyaml