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