RParedesPalacios / cuTensor

Cuda tensors in python

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

cuTensor

Provides a python interface to tensors in GPU. Personal project just to learn how to manage tensors, how to provide a python interface with pybind etc.

TODO

Easier installation with just "pip install"

Install

sudo apt install g++
sudo apt install cmake
sudo apt install nvidia-cuda-toolkit
sudo apt install pybind11-dev 

git clone https://github.com/RParedesPalacios/cuTensor.git

cd cuTensor

python setup.py build_ext --inplace 
pip install .

Test the installation

python -c "import cuTensor"

About

Cuda tensors in python


Languages

Language:C++ 54.7%Language:Cuda 31.7%Language:C 6.8%Language:Python 4.7%Language:CMake 2.1%