diff --git a/docker/requirements-build.txt b/docker/requirements-build.txt index 8c05f3b7bf..648706e362 100644 --- a/docker/requirements-build.txt +++ b/docker/requirements-build.txt @@ -4,6 +4,6 @@ ninja packaging setuptools>=61 setuptools-scm>=8 -torch==2.7.0 +torch==2.7.1 wheel jinja2 diff --git a/docker/requirements-cuda.txt b/docker/requirements-cuda.txt index b2d470b3e9..8c6c13909a 100644 --- a/docker/requirements-cuda.txt +++ b/docker/requirements-cuda.txt @@ -4,7 +4,7 @@ # Dependencies for NVIDIA GPUs ray >= 2.9 nvidia-ml-py # for pynvml package -torch == 2.7.0 +torch == 2.7.1 # These must be updated alongside torch torchvision == 0.19 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version xformers == 0.0.27.post2; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch 2.4.0