RobustVideoMatting icon indicating copy to clipboard operation
RobustVideoMatting copied to clipboard
verified publisher icon PeterL1n

Inference code is not running on my GPU

Open BaccarAla90 opened this issue 3 years ago • 1 comments

Hi, I want to know why the inference is not running on my GPU, in fact, it is only running on my CPU, are there any requirements or drivers I should install for cuda.

BaccarAla90 avatar Jul 14 '22 13:07 BaccarAla90

如果你是在用onnx跑的，那就注意你的模型是cpu版本还是gpu版本的

yuqilol avatar Jan 27 '24 06:01 yuqilol