Unverified Commit b3d445c4 authored by Felix Pultar's avatar Felix Pultar Committed by GitHub
Browse files

Extended compilation instructions C++ API (#157)



* complemented compilation instructions C++ API
Signed-off-by: default avatarFelix Pultar <felix.pultar@phys.chem.ethz.ch>

* complemented compilation instructions C++ API
Signed-off-by: default avatarFelix Pultar <felix.pultar@phys.chem.ethz.ch>

* Update README.md
Signed-off-by: default avatarFelix Pultar <felix.pultar@phys.chem.ethz.ch>
Co-authored-by: default avatarMatthias Fey <matthias.fey@tu-dortmund.de>
parent 3316bdc7
...@@ -284,6 +284,7 @@ pytest ...@@ -284,6 +284,7 @@ pytest
`torch-cluster` also offers a C++ API that contains C++ equivalent of python models. `torch-cluster` also offers a C++ API that contains C++ equivalent of python models.
``` ```
export Torch_DIR=`python -c 'import torch;print(torch.utils.cmake_prefix_path)'`
mkdir build mkdir build
cd build cd build
# Add -DWITH_CUDA=on support for the CUDA if needed # Add -DWITH_CUDA=on support for the CUDA if needed
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment