Skip to content

Commit ce770d8

Browse files
committed
installation readme
1 parent 76bafc3 commit ce770d8

File tree

1 file changed

+39
-0
lines changed

1 file changed

+39
-0
lines changed

README.md

Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,3 +12,42 @@
1212
[![Code Coverage][coverage-image]][coverage-url]
1313

1414
--------------------------------------------------------------------------------
15+
16+
This package consists of a small extension library of optimized sparse matrix operations for the use in [PyTorch](http://pytorch.org/), which are missing and or lack autograd support in the main package.
17+
The package consists of the following methods:
18+
19+
* **[Autograd Sparse Tensor Creation](#Autograd Sparse Tensor Creation)**
20+
* **[Autograd Sparse Tensor Value Extraction](#Autograd Sparse Tensor Value Extraction)**
21+
* **[Sparse Sparse Matrix Multiplication](#Sparse Sparse Matrix Multiplication)**
22+
23+
All included operations work on varying data types and are implemented both for CPU and GPU.
24+
25+
## Installation
26+
27+
Ensure that PyTorch 0.4.0 is installed and verify that `cuda/bin` and `cuda/install` are in your `$PATH` and `$CPATH` respectively, *e.g.*:
28+
29+
```
30+
$ echo $PATH
31+
/usr/local/cuda/bin:...
32+
33+
$ echo $CPATH
34+
/usr/local/cuda/install:...
35+
```
36+
37+
Then run:
38+
39+
```
40+
pip install cffi torch-sparse
41+
```
42+
43+
## Autograd Sparse Tensor Creation
44+
45+
## Autograd Sparse Tensor Value Extraction
46+
47+
## Sparse Sparse Matrix Multiplication
48+
49+
## Running tests
50+
51+
```
52+
python setup.py test
53+
```

0 commit comments

Comments
 (0)