turn WITH_GLTORCH on (#3248)

alibaba · Sep 25, 2023 · a701244 · a701244
1 parent f2cbe07
commit a701244
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -19,7 +19,7 @@ NETWORKX				?= ON
 BUILD_TEST				?= OFF
 
 # whether to build graphlearn-torch extension (graphlearn is built by default)
-WITH_GLTORCH				?= OFF 
+WITH_GLTORCH				?= ON 
 
 # INSTALL_PREFIX is environment variable, but if it is not set, then set default value
 ifeq ($(INSTALL_PREFIX),)

diff --git a/python/requirements.txt b/python/requirements.txt
@@ -24,3 +24,5 @@ tqdm
 click
 vineyard>=0.16.3;sys_platform!="win32"
 simple-parsing
+torch==1.13
+--extra-index-url https://download.pytorch.org/whl/cpu