From aff290a8961af639da5340eb7b40c788d07adead Mon Sep 17 00:00:00 2001 From: fiskrt <43207511+fiskrt@users.noreply.github.com> Date: Thu, 27 Jun 2024 13:05:46 +0200 Subject: [PATCH] Update cpu_requirements.txt --- cpu_requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cpu_requirements.txt b/cpu_requirements.txt index 2f7a5dcd6..aade915be 100644 --- a/cpu_requirements.txt +++ b/cpu_requirements.txt @@ -1,4 +1,4 @@ -# cpu for ci tests +--extra-index-url https://pytorch-geometric.com/whl/torch-1.12.0+cpu.html torch>=1.0,<=1.12.1+cpu; sys_platform == "linux" torch>=1.0,<=1.12.1; sys_platform != "linux" torchvision<=0.13.1+cpu; sys_platform == "linux" @@ -9,3 +9,4 @@ torch-geometric<=2.0.4+cpu; sys_platform == 'linux' torch-geometric<=2.0.4; sys_platform != 'linux' torch-cluster<=1.6.0+cpu; sys_platform == 'linux' torch-cluster<=1.6.0; sys_platform != 'linux' +torch-scatter