diff --git a/pip/cpu_requirements.txt b/pip/cpu_requirements.txt index ff87562f..200f24e6 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.8.0 torch_geometric lightning diff --git a/pip/cuda_requirements.txt b/pip/cuda_requirements.txt index f805072f..c04a0517 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.8.0 torch_geometric lightning diff --git a/pip/docker_requirements.txt b/pip/docker_requirements.txt index 6d1c3d24..7b45a3ed 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.8.0 torch_geometric lightning diff --git a/pip/mchip_requirements.txt b/pip/mchip_requirements.txt index 5fe6dd20..ed23d581 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.8.0 torch_geometric lightning