Ecosyste.ms: Awesome

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

Awesome Lists | Featured Topics | Projects

https://github.com/mjun0812/torch_cpp


https://github.com/mjun0812/torch_cpp

Last synced: about 2 months ago
JSON representation

Awesome Lists containing this project

README

        

# PyTorch C++ Module

PyTorch C++ Module package.

## Install

### wheel install

Download whl file from [releases](https://github.com/mjun0812/torch_cpp/releases) and install it.

```bash
pip install torch_cpp_....whl
```

### source install

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

```bash
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

- [DCNv3](https://github.com/OpenGVLab/InternImage)
- [DCNv4](https://github.com/OpenGVLab/DCNv4)
- [DeformableAttention](https://github.com/fundamentalvision/Deformable-DETR)