Skip to content

Latest commit

 

History

History
30 lines (19 loc) · 468 Bytes

README.md

File metadata and controls

30 lines (19 loc) · 468 Bytes
Raw

#TorchVision ops for libtorch

##Intruction

This is cpp version of torchvision ops for libtorch

##Usage

Linux

mkdir build
cd build
cmake .. -DWITH_CUDA=on

Windows

install MSVC v142 - VS2019 C++ x64/x86 (14.28-16.9)
mkdir build
cd build
cmake .. -DWITH_CUDA=on -Tversion=14.28.29910

link generated libvisionop.so in your project to use torchvision ops, e.g. roi_align, nms.

Here is a example.