Skip to content

Commit d60df4b

Browse files
committed
installation updates
1 parent 0808265 commit d60df4b

File tree

1 file changed

+15
-2
lines changed

1 file changed

+15
-2
lines changed

README.md

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,14 +35,27 @@ All included operations work on varying data types, are implemented both for CPU
3535

3636
## Installation
3737

38-
Check that `nvcc` is accessible from terminal, e.g. `nvcc --version`.
39-
If not, add cuda (`/usr/local/cuda/bin`) to your `$PATH`.
38+
Ensure that at least PyTorch 0.4.1 is installed and verify that `cuda/bin` and `cuda/install` are in your `$PATH` and `$CPATH` respectively, *e.g.*:
39+
40+
```
41+
$ python -c "import torch; print(torch.__version__)"
42+
>>> 0.4.1
43+
44+
$ echo $PATH
45+
>>> /usr/local/cuda/bin:...
46+
47+
$ echo $CPATH
48+
>>> /usr/local/cuda/install:...
49+
```
50+
4051
Then run:
4152

4253
```
4354
pip install torch-scatter
4455
```
4556

57+
If you are running into any installation problems, please create an [issue](https://github.com/rusty1s/pytorch_scatter/issues).
58+
4659
## Example
4760

4861
```py

0 commit comments

Comments
 (0)