An open API service indexing awesome lists of open source software.

https://github.com/pwhiddy/pybind11-cuda

Template for GPU accelerated python libraries
https://github.com/pwhiddy/pybind11-cuda

cuda gpu numpy pybind11 python

Last synced: 10 months ago
JSON representation

Template for GPU accelerated python libraries

Awesome Lists containing this project

README

          

# pybind11-cuda

Compiles out of the box with cmake

Numpy integration

C++ Templating for composable kernels with generic data types

Originally based on https://github.com/torstem/demo-cuda-pybind11

# Prerequisites

Cuda installed in /usr/local/cuda

Python 3.6 or greater

Cmake 3.6 or greater

# To build

```source install.bash```

Test it with
```python3 test_mul.py```