Skip to content

A simple C++14 and CUDA-based header-only library with tools for sparse-machine learning.

Notifications You must be signed in to change notification settings

owensgroup/sparsify.me

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

39 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

git clone https://github.com/neoblizz/sparsify.me.git
cd sparsify.me/examples
make
./bin/sparsify