Skip to content

Commit e60840d

Browse files
committed
update readme [ci skip]
1 parent 3b7b748 commit e60840d

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

README.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,12 +149,13 @@ pytest
149149
## C++ API
150150

151151
`torch-scatter` also offers a C++ API that contains C++ equivalent of python models.
152+
For this, we need to add `TorchLib` to the `-DCMAKE_PREFIX_PATH` (*e.g.*, it may exists in `{CONDA}/lib/python{X.X}/site-packages/torch` if installed via `conda`):
152153

153154
```
154155
mkdir build
155156
cd build
156-
# Add -DWITH_CUDA=on support for the CUDA if needed
157-
cmake ..
157+
# Add -DWITH_CUDA=on support for CUDA support
158+
cmake -DCMAKE_PREFIX_PATH="..." ..
158159
make
159160
make install
160161
```

0 commit comments

Comments
 (0)