https://github.com/owensgroup/sparsify.me
A simple C++14 and CUDA-based header-only library with tools for sparse-machine learning.
https://github.com/owensgroup/sparsify.me
deep-neural-networks deeplearning sparse-matrix sparsification sparsifying-transform sparsity
Last synced: 7 months ago
JSON representation
A simple C++14 and CUDA-based header-only library with tools for sparse-machine learning.
- Host: GitHub
- URL: https://github.com/owensgroup/sparsify.me
- Owner: owensgroup
- Created: 2021-05-04T11:49:36.000Z (over 4 years ago)
- Default Branch: main
- Last Pushed: 2021-08-01T08:19:57.000Z (over 4 years ago)
- Last Synced: 2024-05-08T20:02:36.483Z (over 1 year ago)
- Topics: deep-neural-networks, deeplearning, sparse-matrix, sparsification, sparsifying-transform, sparsity
- Language: C++
- Homepage:
- Size: 7.45 MB
- Stars: 3
- Watchers: 4
- Forks: 0
- Open Issues: 6
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# sparsify.me
Simple header-only library for deep neural network sparse kernels.
## Getting Started
### Dependency: `cusparseLt` for `spmma`
#### Ubuntu 20.04
```
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo apt update
sudo apt-get install libcusparselt0 libcusparselt-dev
```
#### Ubuntu 18.04
```
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt update
sudo apt-get install libcusparselt0 libcusparselt-dev
```
### Quickstart
Needs CUDA version 11 or higher and an CUDA-capable GPU.
```bash
git clone https://github.com/neoblizz/sparsify.me.git
cd sparsify.me/examples
make
./bin/sparsify
```