BackgroundMattingV2 icon indicating copy to clipboard operation
BackgroundMattingV2 copied to clipboard

c++ in gpu

Open luoww1992 opened this issue 4 years ago • 3 comments

my env: win10 vs2019 libtorch 1.7.1 -cuda101-debug cuda101

i can run it by change the device in torch::Device("cpu"); but how to use it in gpu? How did you do it in gpu with torchscript model?

luoww1992 avatar May 21 '21 03:05 luoww1992

Looks like this is more a libtorch issue rather than our model issue. Please try run other models to make sure it is not your environment problem.

PeterL1n avatar May 21 '21 08:05 PeterL1n

@PeterL1n i run it in gpu with c++ sucessfully,but it is slow. i am try to do it. how many fps you test in c++? what should we do to get high fps in c++ gpu with 1920*1080 ?

luoww1992 avatar May 24 '21 05:05 luoww1992

@PeterL1n https://github.com/PeterL1n/BackgroundMattingV2/issues/54#issuecomment-848411458

luoww1992 avatar May 26 '21 02:05 luoww1992