diff --git a/pip/cpu_requirements.txt b/pip/cpu_requirements.txt index ff87562f..39137dd2 100644 --- a/pip/cpu_requirements.txt +++ b/pip/cpu_requirements.txt @@ -7,7 +7,7 @@ tensorflow-cpu~=2.14.0 -torch==2.0.1 +torch==2.7.1 torch_geometric lightning diff --git a/pip/cuda_requirements.txt b/pip/cuda_requirements.txt index f805072f..526c28e3 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 torch_geometric lightning diff --git a/pip/docker_requirements.txt b/pip/docker_requirements.txt index 6d1c3d24..51ab030f 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 torch_geometric lightning diff --git a/pip/mchip_requirements.txt b/pip/mchip_requirements.txt index 5fe6dd20..b1a3c45a 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 torch_geometric lightning