diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index 5b1eca2c89..7b96066dc9 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -3,8 +3,8 @@ tensorflow-cpu>=2.16.1 # Torch with cuda support. --extra-index-url https://download.pytorch.org/whl/cu121 -torch==2.4.0+cu121 -torchvision==0.19.0+cu121 +torch==2.5.1+cu121 +torchvision==0.20.1+cu121 # Jax cpu-only version. jax[cpu]