diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index 3b34f32aac..eeb54652d6 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -4,8 +4,8 @@ tensorflow-text~=2.17 # Torch with cuda support. --extra-index-url https://download.pytorch.org/whl/cu121 -torch==2.3.0+cu121 -torchvision==0.18.0+cu121 +torch==2.4.0+cu121 +torchvision==0.19.0+cu121 # Jax cpu-only version. jax[cpu]