/torch_cpp

Primary LanguageCuda

PyTorch C++ Module

PyTorch C++ Module package.

Install

wheel install

Download whl file from releases and install it.

pip install torch_cpp_....whl

source install

Before installing this repository, you need to install torch and CUDA to build module.

pip install torch

# Source Install
git clone https://github.com/mjun0812/torch_cpp.git
cd torch_cpp
python setup.py build install --user

# pip install from github
pip install git+https://github.com/mjun0812/torch_cpp.git

Provided Module