Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-sparse
Commits
ce770d81
Commit
ce770d81
authored
Jul 31, 2018
by
rusty1s
Browse files
installation readme
parent
76bafc30
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
39 additions
and
0 deletions
+39
-0
README.md
README.md
+39
-0
No files found.
README.md
View file @
ce770d81
...
...
@@ -12,3 +12,42 @@
[
![Code Coverage
][
coverage-image
]
][coverage-url]
--------------------------------------------------------------------------------
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.
The package consists of the following methods:
*
**[Autograd Sparse Tensor Creation](#Autograd Sparse Tensor Creation)**
*
**[Autograd Sparse Tensor Value Extraction](#Autograd Sparse Tensor Value Extraction)**
*
**[Sparse Sparse Matrix Multiplication](#Sparse Sparse Matrix Multiplication)**
All included operations work on varying data types and are implemented both for CPU and GPU.
## Installation
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.*
:
```
$ echo $PATH
/usr/local/cuda/bin:...
$ echo $CPATH
/usr/local/cuda/install:...
```
Then run:
```
pip install cffi torch-sparse
```
## Autograd Sparse Tensor Creation
## Autograd Sparse Tensor Value Extraction
## Sparse Sparse Matrix Multiplication
## Running tests
```
python setup.py test
```
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment