diff --git a/pip/cpu_requirements.txt b/pip/cpu_requirements.txt index 4e7940e3..92bc85fe 100644 --- a/pip/cpu_requirements.txt +++ b/pip/cpu_requirements.txt @@ -20,7 +20,7 @@ # ============================= # PyTorch CPU; Python 3.10 compatible; aligned with DGL torch-2.1 wheels -torch==2.1.2+cpu +torch==2.7.1+cpu torchdata==0.7.1 # Lightning; pin to a recent stable series to avoid breaking changes diff --git a/pip/cuda_requirements.txt b/pip/cuda_requirements.txt index c8b753db..11b9ef15 100644 --- a/pip/cuda_requirements.txt +++ b/pip/cuda_requirements.txt @@ -9,7 +9,7 @@ tensorflow[and-cuda]~=2.14.0 tensorrt -torch==2.0.1 +torch==2.7.1+cpu torch_geometric lightning diff --git a/pip/docker_requirements.txt b/pip/docker_requirements.txt index 24d9de2e..6357cd97 100644 --- a/pip/docker_requirements.txt +++ b/pip/docker_requirements.txt @@ -7,7 +7,7 @@ tensorflow-cpu~=2.14.0 -torch==2.0.1 +torch==2.7.1+cpu torch_geometric lightning diff --git a/pip/mchip_requirements.txt b/pip/mchip_requirements.txt index 7b44d6fa..91e38cc6 100644 --- a/pip/mchip_requirements.txt +++ b/pip/mchip_requirements.txt @@ -8,7 +8,7 @@ # use tensorflow not tensorflow-cpu for arm. see https://www.tensorflow.org/install/pip#linux tensorflow~=2.14.0 -torch==2.0.1 +torch==2.7.1+cpu torch_geometric lightning