Update cpu_requirements.txt

GT4SD · Jun 27, 2024 · aff290a · aff290a
1 parent 4daf402
commit aff290a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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