diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index b8ff9cca51..cb5affc125 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -4,8 +4,8 @@ tensorflow-text~=2.18 # Torch with cuda support. --extra-index-url https://download.pytorch.org/whl/cu126 -torch==2.6.0+cu126 -torchvision==0.21.0+cu126 +torch==2.7.0+cu126 +torchvision==0.22.0+cu126 # Jax cpu-only version. jax[cpu]