myay/PyTorch-CUDA-Example

A minimal template for custom CUDA extensions in Pytorch.

A minimal example for PyTorch CUDA extensions.
Simplified version of https://github.com/pytorch/extension-cpp.

Install requirements:
CUDA toolkit, PyTorch, pybind11

To install:
python setup.py install

To run:
python test.py