diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 3eb91212e976..c4e05694ab36 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -4,7 +4,7 @@ # Dependencies for NVIDIA GPUs ray >= 2.9 nvidia-ml-py # for pynvml package -torch == 2.3.1 +torch==2.8.0 # These must be updated alongside torch torchvision == 0.18.1 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version xformers == 0.0.27 # Requires PyTorch 2.3.1