smrfeld / pytorch-cpp-tutorial

Tutorial for Python package calling a custom PyTorch C++ module

Tutorial for Python package calling a custom PyTorch C++ module

This is a minimal example of a Python package calling a custom PyTorch C++ module.

Installing & running

0. (Optional) Create a conda environment:

   conda create -n test-pytorch-cpp python=3.11
   conda activate test-pytorch-cpp

1. Install requirements:

   pip install -r requirements.txt

2. Install package using setup.py:

   pip install -e .

3. Run the test:

   python main.py

   Expected result:

   tensor([5., 7., 9.])